在满足条件之前过滤掉所有元素，之后保留所有元素

Question

我想知道下面的问题是否有一个简单的解决方案。这里的问题是，我希望在初始条件为真后保留列表中出现的每个元素。这里的条件是，我想删除值大于18的条件之前的所有内容，但保留之后的所有内容。示例

输入：

p = [4,9,10,4,20,13,29,3,39]

预期产出：

p = [20,13,29,3,39]

我知道你可以通过

[x for x in p if x>18] 

但是我想一旦找到18以上的第一个值就停止这个操作，然后包括其余的值，不管它们是否满足条件。这似乎是一个简单的问题，但我还没有找到解决办法。

Answer 1

伟大的解决方案在这里，只是想演示如何做到这一点与Numpy：

>>> import numpy as np
>>> p[(np.array(p) > 18).argmax():]
[20, 13, 29, 3, 39]

因为这里有很多很好的答案，所以我决定运行一些简单的基准测试。第一个使用OP的示例数组（[4,9,10,4,20,13,29,3,39]），长度为9。第二个使用随机生成的长度2万数组，其中前半部分在0到15之间，后半部分在-20到30之间（这样分裂就不会发生在中心）。

使用OP的数据（长度为9的数组）：

%timeit enke()
650 ns ± 15.9 ns per loop (mean ± std. dev. of 7 runs, 1000000 loops each)

%timeit j1lee1()
546 ns ± 4.22 ns per loop (mean ± std. dev. of 7 runs, 1000000 loops each)

%timeit j1lee2()
551 ns ± 19 ns per loop (mean ± std. dev. of 7 runs, 1000000 loops each)

%timeit j2lee3()
536 ns ± 12.9 ns per loop (mean ± std. dev. of 7 runs, 1000000 loops each)

%timeit richardec()
2.08 µs ± 16 ns per loop (mean ± std. dev. of 7 runs, 100000 loops each)

使用长度为20000（20000）的数组：

%timeit enke()
1.5 ms ± 34.5 µs per loop (mean ± std. dev. of 7 runs, 1000 loops each)

%timeit j1lee1()
1.95 ms ± 43 µs per loop (mean ± std. dev. of 7 runs, 1000 loops each)

%timeit j1lee2()
2.1 ms ± 53.7 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)

%timeit j2lee3()
2.33 ms ± 96.2 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)

%timeit richardec()
13.3 µs ± 461 ns per loop (mean ± std. dev. of 7 runs, 100000 loops each)

生成第二个数组的代码：

p = np.hstack([np.random.randint(0,15,10000),np.random.randint(-20,30,10000)])

所以，对于小情况，Numpy是一个鼻涕虫，不需要。但大的情况下，Numpy几乎是100x倍的速度和方式去！：）

Answer 2

您可以使用枚举并在下一步中列出切片：

out = next((p[i:] for i, item in enumerate(p) if item > 18), [])

输出：

[20, 13, 29, 3, 39]

Answer 3

你可以使用itertools。dropwhile：

import itertools

p = [4,9,10,4,20,13,29,3,39]

p = list(itertools.dropwhile(lambda x: x <= 18, p))
print(p) # [20, 13, 29, 3, 39]

使用walrus操作符（不使用itertools，但仅在python 3.8中可用）：

exceeded = False
p = [x for x in p if (exceeded := exceeded or x > 18)]
print(p) # [20, 13, 29, 3, 39]

但我猜有些人不喜欢这种风格。在这种情况下，您可以将其展开为：

output = []
exceeded = False
for x in p:
    exceeded = exceeded or x > 18
    if exceeded:
        output.append(x)

print(output) # [20, 13, 29, 3, 39]