python - 根据过滤器从python列表中删除元素

Question

我有 3 个 Python 列表。

list_a = [10., 20., 30., 12.]
list_b = [30., 20., 60., 12.]
list_c = [10., 80., 90., 12.]

我想将这些元素放入 list_b 中和 list_c list_a 中的值是 <= 15 .因此结果变为:

list_b = [20., 60.]
list_c = [80., 90.]

有没有不用 for 循环的方法？ (列表理解没问题)

Answer 1

如果你像你说的那样使用 numpy，在注释中，你可以简单地从 a 创建一个 bool 索引，并屏蔽 b 和 c 中的元素:

import numpy as np

list_a = np.array([10., 20., 30., 12.])
list_b = np.array([30., 20., 60., 12.])
list_c = np.array([10., 80., 90., 12.])

list_b = list_b[list_a>15]
list_c = list_c[list_a>15]

print list_a
print list_b
print list_c

输出:

[ 10.  20.  30.  12.]
[ 20.  60.]
[ 80.  90.]

您可以通过.tolist() 方法将lists_b 和list_c 转换为Python list type。