从python中的列表创建生成器表达式答案

【问题标题】：Creating a generator expression from a list in python从python中的列表创建生成器表达式
【发布时间】：2009-12-04 23:00:35
【问题描述】：

在 Python 中执行以下操作的最佳方法是什么：

for item in [ x.attr for x in some_list ]:
    do_something_with(item)

这可能是个小问题，但列表推导式不是生成了一个我们不需要的新列表，只是占用了内存吗？如果我们能做一个类似迭代器的列表理解不是更好吗？

【问题讨论】：

【解决方案1】：

是的（对你的两个问题）。

通过使用圆括号而不是方括号，您可以为该序列创建所谓的“生成器表达式”，这完全符合您的建议。它使您可以遍历序列，而无需分配一个列表来同时保存所有元素。

for item in (x.attr for x in some_list):
    do_something_with(item)

生成器表达式的详细信息记录在PEP 289。

【讨论】：

【解决方案2】：

为什么不只是：

for x in some_list:
    do_something_with(x.attr)

【讨论】：

【解决方案3】：

这个问题被标记为functional-programming 没有适当的答案，所以这里有一个实用的解决方案：

from operator import itemgetter

map(do_something_with, map(itemgetter('attr'), some_list))

Python 3 的 map() 使用迭代器，但 Python 2 会创建一个列表。对于 Python 2，请改用 itertools.imap()。

如果您返回 some_list，您可以使用生成器表达式和惰性求值进一步简化它：

def foo(some_list):
    return (do_something_with(item.attr) for item in some_list)

【讨论】：