Pythonic 从一个列表弹出到另一个列表

Question

我有两个清单a = [1,2,3]b = []如果满足特定条件，我想从列表 a 中移动一个元素。a = [1,3]b = [2]下面的代码显示了一个示例，但是，我想在单个循环内执行此操作。我如何更有效地做到这一点？a = [1,2,3]b = []pop_list = []for i in range(len(a)):    if a[i] == 2:        print("pop:", a[i])        pop_list.append(i)for i in range(len(pop_list)):    b.append(a.pop(pop_list[i]))# Reset pop_listpop_list=[]理想情况下，我不会生成新列表 b。

Answer 1

一对列表推导式即可完成这项工作：一个用于选择 所需的元素b，另一个用于从中删除它们a

b = [i for i in a if i == 2]

a = [i for i in a if i != 2]

Answer 2

您可以对filter两者itertools.filterfalse使用相同的过滤函数：

from itertools import filterfalse

a = [1,2,3]

b = []

list(filterfalse(lambda x: x == 2, a))

list(filter     (lambda x: x == 2, a))

[1, 3]

[2]

Answer 3

如果该元素x存在，您可以将其从 中删除b并将其附加到a.

a = [1, 2, 3]

b = []

x = 2

def remove_append(a, b, x):

    if x in a:

        a.remove(x)

        b.append(x)

remove_append(a, b, x)

print(a)

print(b)

输出：

[1, 3]

[2]

Answer 4

我们必须遍历所有元素，但是，您可以应用此技巧在一个循环中添加到适当的列表：

（追加到循环比删除任意位置的元素更有效）

a = [1,2,3]

condition_false, condition_true = [], []

for v in a:

    # Add to the right list

    (condition_false, condition_true)[v == 2].append(v)

# [1, 3]

print(condition_false)

# [2]

print(condition_true)