Python:打印一个生成器表达式?

时间:2011-03-02 07:34:17

标签: python

在Python shell中,如果我输入列表推导,例如:

>>> [x for x in string.letters if x in [y for y in "BigMan on campus"]]

我得到了一个很好的印刷结果:

['a', 'c', 'g', 'i', 'm', 'n', 'o', 'p', 's', 'u', 'B', 'M']

字典理解相同:

>>> {x:x*2 for x in range(1,10)}
{1: 2, 2: 4, 3: 6, 4: 8, 5: 10, 6: 12, 7: 14, 8: 16, 9: 18}

如果我输入生成器表达式,我得不到如此友好的回应:

>>> (x for x in string.letters if x in (y for y in "BigMan on campus"))
<generator object <genexpr> at 0x1004a0be0>

我知道我可以这样做:

>>> for i in _: print i,
a c g i m n o p s u B M

除此之外(或编写辅助函数),我可以在交互式shell中轻松评估和打印该生成器对象吗?

5 个答案:

答案 0 :(得分:135)

快速回答:

围绕生成器表达式执行list()(几乎)完全等同于围绕它的[]括号。所以是的,你可以做到

>>> list((x for x in string.letters if x in (y for y in "BigMan on campus")))

但你也可以这样做

>>> [x for x in string.letters if x in (y for y in "BigMan on campus")]

是的,这会将生成器表达式转换为列表理解。它是相同的东西,并在其上调用list()。 因此,将生成器表达式放入列表的方法是在其周围放置括号。

详细说明:

生成器表达式是“裸”for表达式。像这样:

x*x for x in range(10)

现在,你不能将它自己粘在一条线上,你会得到一个语法错误。但是你可以在它周围添加括号。

>>> (x*x for x in range(10))
<generator object <genexpr> at 0xb7485464>

这有时被称为生成器理解,虽然我认为官方名称仍然是生成器表达式,但实际上没有任何区别,括号只是为了使语法有效。如果您将其作为函数的唯一参数传递给函数,则不需要它们,例如:

>>> sorted(x*x for x in range(10))
[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

基本上,Python 3和Python 2.7中提供的所有其他理解只是围绕生成器表达式的语法糖。设置理解:

>>> {x*x for x in range(10)}
{0, 1, 4, 81, 64, 9, 16, 49, 25, 36}

>>> set(x*x for x in range(10))
{0, 1, 4, 81, 64, 9, 16, 49, 25, 36}

Dict comprehensions:

>>> dict((x, x*x) for x in range(10))
{0: 0, 1: 1, 2: 4, 3: 9, 4: 16, 5: 25, 6: 36, 7: 49, 8: 64, 9: 81}

>>> {x: x*x for x in range(10)}
{0: 0, 1: 1, 2: 4, 3: 9, 4: 16, 5: 25, 6: 36, 7: 49, 8: 64, 9: 81}

Python 3下的列表推导:

>>> list(x*x for x in range(10))
[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

>>> [x*x for x in range(10)]
[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

在Python 2下,列表推导不仅仅是语法糖。但唯一的区别是x将在Python 2下泄漏到名称空间。

>>> x
9

在Python 3下你会得到

>>> x
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
NameError: name 'x' is not defined

这意味着在Python中获得生成器表达式内容的良好打印输出的最佳方法是从列表中理解它!但是,如果您已经有一个生成器对象,这显然不起作用。这样做只会列出一个生成器:

>>> foo = (x*x for x in range(10))
>>> [foo]
[<generator object <genexpr> at 0xb7559504>]

在这种情况下,您需要致电list()

>>> list(foo)
[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

虽然这有效,但有点愚蠢:

>>> [x for x in foo]
[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

答案 1 :(得分:15)

您可以将表达式包装在对list

的调用中
>>> list(x for x in string.letters if x in (y for y in "BigMan on campus"))
['a', 'c', 'g', 'i', 'm', 'n', 'o', 'p', 's', 'u', 'B', 'M']

答案 2 :(得分:12)

与列表或字典不同,生成器可以是无限的。这样做不会起作用:

def gen():
    x = 0
    while True:
        yield x
        x += 1
g1 = gen()
list(g1)   # never ends

此外,阅读发电机会改变它,因此不是一种完美的观察方式。 要查看生成器输出的示例,您可以执行

g1 = gen()
[g1.next() for i in range(10)]

答案 3 :(得分:9)

或者您可以始终map通过迭代器,而无需构建中间列表:

>>> _ = map(sys.stdout.write, (x for x in string.letters if x in (y for y in "BigMan on campus")))
acgimnopsuBM

答案 4 :(得分:2)

>>> list(x for x in string.letters if x in (y for y in "BigMan on campus"))
['a', 'c', 'g', 'i', 'm', 'n', 'o', 'p', 's', 'u', 'B', 'M']