请教一个关于 map 和 itertools.chain 嵌套的问题

2017-08-16 13:36:33 +08:00
 quinoa42

代码如下:

import itertools
flaten = itertools.chain.from_iterable # 方便起见

def a(i):
    print("a", i)
    return list(range(i))
def b(i):
    print("b", i)
    return "+"*i
    
list(map(b,flaten((map(a, [1,2,3,4,5])))))

在解释器下输出为:

a 1
b 0
a 2
b 0
b 1
a 3
b 0
b 1
b 2
a 4
b 0
b 1
b 2
b 3
a 5
b 0
b 1
b 2
b 3
b 4
Out[6]:
['',
 '',
 '+',
 '',
 '+',
 '++',
 '',
 '+',
 '++',
 '+++',
 '',
 '+',
 '++',
 '+++',
 '++++']

我之前以为会像嵌套的函数一样先执行内层再执行外层(即输出为a a a a a ... b b b b b...),结果发现似乎并非如此。我的理解为,实际上的过程似乎是在 calllist()的时候才执行。在map(b, ...)第一次调用chain对象的__next__时,chain调用了内层的map__next__,而这又返回了a(i)的值(一个list)。而接下来的几次chain.__next__都只是从这个list中取值直到用尽,用尽后再次调用内层的map.__next__

请问这样的理解是否正确?

2096 次点击
所在节点    Python
2 条回复
guyskk
2017-08-16 15:08:43 +08:00
Python3 里面 map/range/flaten... 返回的都是生成器,不迭代它它就不会执行
gemini
2017-08-16 15:13:13 +08:00
楼上正解

这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。

https://www.v2ex.com/t/383361

V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。

V2EX is a community of developers, designers and creative people.

© 2021 V2EX