python函数产生元组，只需要一个元素

Question

我有一个函数

def f():
    # whatever
    yield (a,b)

现在我想收集所有的 a 但不是 b。我还希望结果 aa 是一个列表而不是迭代器。现在我用

aa, _ = zip(*f())

就空间/时间效率而言，这是最好的吗？

Answer 1

zip(*seq) 必须摄取整个生成器，然后才能输出列。这效率不高。

只坚持列表理解。您可以使用元组赋值:

aa = [a for a, _ in f()]

或使用索引:

aa = [tup[0] for tup in f()]

如果您没有让所有值都可用于随机访问或其他必须有列表的操作，您可以使用生成器表达式来保持内存效率:

aa = (a for a, _ in f())