在 Python 中使用列表理解来执行类似于 zip() 的操作？

Question

我是一名 Python 新手，我想做的一件事就是围绕列表理解进行思考。我可以看到这是一个非常强大的功能，值得学习。

cities = ['Chicago', 'Detroit', 'Atlanta']
airports = ['ORD', 'DTW', 'ATL']

print zip(cities,airports)
[('Chicago', 'ORD'), ('Detroit', 'DTW'), ('Atlanta', 'ATL')]

如何使用列表理解，以便可以将结果作为列表中的一系列列表而不是列表中的一系列元组来获取？

[['Chicago', 'ORD'], ['Detroit', 'DTW'], ['Atlanta', 'ATL']]

（我意识到字典在这种情况下可能更合适，但我只是想更好地理解列表）。谢谢！

Answer 1

像这样的东西：

[[c, a] for c, a in zip(cities, airports)]

或者，list构造函数可以将元组转换为列表：

[list(x) for x in zip(cities, airports)]

或者，在这种情况下，map函数稍微不那么冗长：

map(list, zip(cities, airports))

Answer 2

如果您想完全不使用 zip 来完成此操作，则必须执行以下操作：

[ [cities[i],airports[i]] for i in xrange(min(len(cities), len(airports))) ]

但除了智力练习之外，没有理由这样做。

使用 map(list, zip(cities, airports)) 更短、更简单，而且几乎肯定会运行得更快。

Answer 3

如果没有 zip、map 或 itertools 的帮助，列表理解无法在多个序列上建立“并行循环”——只能是简单的在一个序列上循环，或在多个序列上“嵌套”循环。

Answer 4

这需要 zip 的输出并将所有元组转换为列表：

map(list, zip(cities, airports))

至于每个元组的性能：

$ python -m timeit -c '[ [a, b] for a, b in zip(xrange(100), xrange(100)) ]'
10000 loops, best of 3: 68.3 usec per loop

$ python -m timeit -c 'map(list, zip(xrange(100), xrange(100)))'
10000 loops, best of 3: 75.4 usec per loop

$ python -m timeit -c '[ list(x) for x in zip(range(100), range(100)) ]'
10000 loops, best of 3: 99.9 usec per loop

Answer 5

也可以使用enumerate：

[[y,airports[x]] for x,y in enumerate(cities)]

Answer 6

如果您想使用列表理解，并创建一个变量作为字典，这里是一个不错的方法：

newvariable = {key: value for key, value in zip(cities,airports)}

Answer 7

您可以使用列表理解来组合两个列表，但在现实世界中，您应该更喜欢内置函数 (zip()) 而不是复杂的流语句。

zip_alternative=[ (cities[x],airports[y]) for x  in range(len(cities)) for y  in range(len(airports)) if x==y ]