Python在列表理解中使用枚举

Posted 2023-02-22

【中文标题】Python在列表理解中使用枚举

【英文标题】：Python using enumerate inside list comprehension

【发布时间】：2012-06-02 09:14:50

【问题描述】：

假设我有一个这样的列表：

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

所以，我的问题是：在列表理解中使用 enumerate 的正确方法是什么？

【参考方案1】：

试试这个：

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

您需要将i,j 放在一个元组中，列表解析才能工作。或者，鉴于enumerate()已经返回了一个元组，你可以直接返回它而无需先解包：

[pair for pair in enumerate(mylist)]

无论哪种方式，返回的结果都符合预期：

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

【讨论】：

不是必须是元组。任何使用 i 和 j 返回值的表达式都可以

枚举的使用很棒，使用itertools如何提高效率

请记住，元组是由, 而不是 () 构建的。所以“将i,j 放在一个元组中”是没有意义的，因为i,j 已经是一个元组了！问题是列表比较解析器需要语句组的括号。

【参考方案2】：

明确一点，这与 enumerate 无关，与列表解析语法有关。

这个列表推导返回一个元组列表：

[(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']

这是不一致的（恕我直言）并且与字典理解语法混淆：

>>> 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')])

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

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

【参考方案3】：

或者，如果您不坚持使用列表推导：

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

【参考方案4】：

这是一种方法：

>>> 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 周围的 () 以使其成为元组。

【参考方案5】：

如果您使用的是长列表，则列表理解似乎更快，更不用说更具可读性了。

~$ 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

【讨论】：

+1 我没有测试它，但我敢打赌[t for t in enumerate(my list)] 会更快。

【参考方案6】：

要明确说明元组。

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

【参考方案7】：

所有答案都很棒的家伙。我知道这里的问题是针对枚举的，但是像这样的事情怎么样，只是另一个角度

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)