python - 如何根据条件拆分列表？

Question

目前我还没有找到一个方便的方法来按特定条件拆分列表，例如，我有一个记录列表:

 a = ((0,1),(1,0),(0,2),(1,0),(3,0),(4,0),(0,3),(1,5)....)

我想将内容分成 2 个列表

alist = []
blist = []
for x in a:
    if x[0] == 0:
        alist.append(x)
    elif x[0] == 1:
        blist.append(x)

不是很简洁。

写成列表推导式:

aList = [x for x in a if x[0] == 0]
bList = [x for x in a if x[0] == 1]

列表理解通常有利于阅读和性能，但在这种情况下，列表必须迭代两次。

有没有更好的方法来完成这项工作？

Answer 1

添加一行将使循环更加简洁，但代价是可读性(和 FPness)。

alist = []
blist = []
bothlists = [alist, blist]
for x in a:
  bothlists[x[0]].append(x)