如何使用Python 2.7在itertools中提供zip_longest

Question

尝试在Windows 10上运行的Python Jupyter 2.7 nb上导入此函数时，出现此错误：

我相信过去我没有遇到任何问题，因为我使用的是Python 3.所以我想知道它是否只是在Python 2中不可用，或者是否有办法让它工作。

Answer 1

对于Python 3，方法为zip_longest：

from itertools import zip_longest

对于Python 2，方法是izip_longest：

from itertools import izip_longest

Answer 2

如果您不知道哪个版本的python运行脚本，您可以使用此技巧：

try:
    from itertools import zip_longest as zip_longest
except:
    from itertools import izip_longest as zip_longest

# now this works in both python 2 and 3
print(list(zip_longest([1,2,3],[4,5])))