python - 使用元组过滤和拆包处理空情况

Question

我有一些并行列表需要根据其中一个列表中的值进行过滤。有时我会写这样的东西来过滤它们:

lista = [1, 2, 3]
listb = [7, 8, 9]
filtered_a, filtered_b = zip(*[(a, b) for (a, b) in zip(lista, listb) if a < 3])

这给出了 filtered_a == (1, 2)和 filtered_b == (7, 8)

但是，将最终条件从 a < 3 更改为至 a < 0引发异常:

Traceback (most recent call last):
  ...
ValueError: need more than 0 values to unpack

我知道为什么会这样:列表理解是空的，所以就像调用 zip(*[]) ，类似于 zip() ，它只返回一个空列表，该列表不能解压缩到单独的 filtered_a 和 filtered_b 迭代器中。

是否有更好的(更短、更简单、更 pythonic)处理空案例的过滤函数？在空的情况下，我希望 filtered_a 和 filtered_b 是空的可迭代对象，因此任何后续代码都可以保持不变。

Answer 1

您可以使用默认值简单地短路:

filtered_a, filtered_b = zip(*[(a, b) for a, b in zip(lista, listb) if a < 0]) or ([], [])
print(filtered_b, filtered_a)
# [] []

对于 Python 3，您需要在 zip 返回的迭代器上调用 list，以便第一个操作数可以被评估为空列表(不是迭代器) ，否则即使迭代器可能为空，也永远不会达到默认值，因为 bool(iter([])) 为 True。