关于嵌套生成器理解字典的误解

时间:2012-10-09 01:50:06

标签: python dictionary generator list-comprehension

我尝试使用嵌套的生成器理解字典,将列表作为存储值,并观察以下奇怪的(对我而言)行为:

Python 2.6.5 (r265:79063, Oct  1 2012, 22:07:21) 
[GCC 4.4.3] on linux2
Type "help", "copyright", "credits" or "license" for more information.
>>> dummy = {1:[1,2,3],2:[2,3,4],3:[3,4,5]}
>>> a = (d for _,d in dummy.iteritems())
>>> a.next()
[1, 2, 3]
>>> a.next()
[2, 3, 4]
>>> a.next()
[3, 4, 5]
>>> a.next()
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
StopIteration

这是有道理的。以下内容(至少对我而言)

>>> aa = (dd for dd in (d for _,d in dummy.iteritems()))
>>> aa.next()
[1, 2, 3]
>>> aa.next()
[2, 3, 4]
>>> aa.next()
[3, 4, 5]
>>> aa.next()
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
StopIteration

为什么我的(尝试的)嵌套生成器理解的行为与非嵌套版本的行为相同?我希望每个aa.next()给出一个单独的元素结果,而不是列表。

3 个答案:

答案 0 :(得分:1)

每次迭代时,内部生成器都会返回一个值,每个值都是一个列表。您将需要使用实际的嵌套结构。

>>> aa = (d3 for dd in (d for _,d in dummy.iteritems()) for d3 in dd)
>>> next(aa)
1
>>> next(aa)
2
>>> next(aa)
3
>>> next(aa)
2
>>> next(aa)
3
>>> next(aa)
4
>>> next(aa)
3
>>> next(aa)
4
>>> next(aa)
5
>>> next(aa)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
StopIteration

答案 1 :(得分:1)

要进行嵌套,您不需要内部生成器。相反,使用普通的嵌套for循环:

>>> dummy = {1:[1,2,3],2:[2,3,4],3:[3,4,5]}
>>> aa = (dd for _,d in dummy.iteritems() for dd in d)
>>> list(aa)
[1, 2, 3, 2, 3, 4, 3, 4, 5]

这将与:

相同
def aa(dummy):
    for _, d in dummy.iteritems():
        for dd in d:
            yield dd

print list(aa())

学习生成器表达式的最佳方法是从普通生成器开始,这样您就可以在每个步骤中轻松查看结果。

答案 2 :(得分:0)

这不是真正的“嵌套”生成器,正如你在想的那样。

您的代码(大致)等同于:

for dd in (d for _,d in dummy.iteritems()):
    print dd

通常,“嵌套”用于表示类似的内容:

(dd for _, d in dummy.iteritems() for dd in d)