在列表内涵中使用枚举

让我们假设我有一个这样的列表:

mylist = ["a","b","c","d"]

为了得到与索引一起打印的值，我可以像这样使用 Python 的 enumerate函数

>>> for i,j in enumerate(mylist):
...     print i,j
...
0 a
1 b
2 c
3 d
>>>

现在，当我尝试在 list comprehension中使用它时，它会给我这个错误

>>> [i,j for i,j in enumerate(mylist)]
File "<stdin>", line 1
[i,j for i,j in enumerate(mylist)]
^
SyntaxError: invalid syntax

因此，我的问题是: 在列表内涵中使用枚举的正确方法是什么？

279771 次浏览
小开
最佳答案

试试这个:

[(i, j) for i, j in enumerate(mylist)]

你需要把 abc 0放在一个 tuple 中，这样列表内涵才能工作。或者，假设 enumerate() 已经返回一个元组，您可以直接返回它，而不必首先解压缩它:

[pair for pair in enumerate(mylist)]

不管是哪种情况，返回的结果都与预期的一样:

> [(0, 'a'), (1, 'b'), (2, 'c'), (3, 'd')]
小开

这里有一个方法:

>>> mylist = ['a', 'b', 'c', 'd']
>>> [item for item in enumerate(mylist)]
[(0, 'a'), (1, 'b'), (2, 'c'), (3, 'd')]

或者，你可以:

>>> [(i, j) for i, j in enumerate(mylist)]
[(0, 'a'), (1, 'b'), (2, 'c'), (3, 'd')]

出现错误的原因是，为了使它成为元组，在 i和 j周围缺少()。

小开

清楚地说明元组。

[(i, j) for (i, j) in enumerate(mylist)]
小开

或者，如果你不坚持使用列表内涵:

>>> mylist = ["a","b","c","d"]
>>> list(enumerate(mylist))
[(0, 'a'), (1, 'b'), (2, 'c'), (3, 'd')]
小开

只是想弄清楚，这与 enumerate没有任何关系，而是与列表内涵语法有关。

这个列表内涵返回一个 tuple 列表:

[(i,j) for i in range(3) for j in 'abc']

这是一系列的词汇:

[{i:j} for i in range(3) for j in 'abc']

一份清单:

[[i,j] for i in range(3) for j in 'abc']

语法错误:

[i,j for i in range(3) for j in 'abc']

这是不一致的(IMHO) ，并且与字典理解语法混淆:

>>> {i:j for i,j in enumerate('abcdef')}
{0: 'a', 1: 'b', 2: 'c', 3: 'd', 4: 'e', 5: 'f'}

还有一组元组:

>>> {(i,j) for i,j in enumerate('abcdef')}
set([(0, 'a'), (4, 'e'), (1, 'b'), (2, 'c'), (5, 'f'), (3, 'd')])

正如 Ócar López 所说，您可以直接传递枚举元组:

>>> [t for t in enumerate('abcdef') ]
[(0, 'a'), (1, 'b'), (2, 'c'), (3, 'd'), (4, 'e'), (5, 'f')]
小开

如果你使用的是长列表，那么这个列表内涵看起来更快，更不用说它的可读性了。

~$ python -mtimeit -s"mylist = ['a','b','c','d']" "list(enumerate(mylist))"
1000000 loops, best of 3: 1.61 usec per loop
~$ python -mtimeit -s"mylist = ['a','b','c','d']" "[(i, j) for i, j in enumerate(mylist)]"
1000000 loops, best of 3: 0.978 usec per loop
~$ python -mtimeit -s"mylist = ['a','b','c','d']" "[t for t in enumerate(mylist)]"
1000000 loops, best of 3: 0.767 usec per loop
小开

都是很棒的答案家伙。我知道这里的问题是特定于枚举的，但是像这样的东西怎么样，只是另一个角度

from itertools import izip, count
a = ["5", "6", "1", "2"]
tupleList = list( izip( count(), a ) )
print(tupleList)

如果需要在性能方面并行迭代多个列表，那么它会变得更加强大

a = ["5", "6", "1", "2"]
b = ["a", "b", "c", "d"]
tupleList = list( izip( count(), a, b ) )
print(tupleList)

提交答案