Python：在地图对象上调用'list'两次

Question

我想计算直到n的平方和。假设n为4.然后，此代码生成一个列表，其中包含0到4范围内的地图对象：

m = map(lambda x: x**2, range(0,4))

足够轻松。现在在m上调用list，然后求和：

>>> sum(list(m))
14

意外的行为是，如果我再次运行最后一行，则总和为0：

>>> sum(list(m))
0

我怀疑这是因为调用list(m)会返回一个空列表，但我无法找到此行为的解释。有人可以帮我解决这个问题吗？

Answer 1

map在Python 3中返回有状态迭代器。状态迭代器可能只被消耗一次，之后它就会耗尽并且不会产生任何值。

在您的代码段中，您多次消耗迭代器。 list(m)每次尝试重新创建列表，对于第二次和下次运行，创建的列表将始终为空（因为源迭代器已在第一次list(m)操作中使用）。

只需将迭代器转换为列表一次，然后在上述列表中操作。

m = map(lambda x: x**2, range(0,4))
l = list(m)
assert sum(l) == 14
assert sum(l) == 14