假设我有一个这样的列表:
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
所以,我的问题是:在列表理解中使用枚举的正确方法是什么?
答案 0 :(得分:149)
试试这个:
[(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')]
答案 1 :(得分:44)
只是要非常清楚,这与enumerate
无关,而与列表理解语法有关。
此列表推导返回元组列表:
[(i,j) for i in range(3) for j in 'abc']
这是一个dicts列表:
[{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')])
正如ÓscarLópez所说,你可以直接传递枚举元组:
>>> [t for t in enumerate('abcdef') ]
[(0, 'a'), (1, 'b'), (2, 'c'), (3, 'd'), (4, 'e'), (5, 'f')]
答案 2 :(得分:31)
或者,如果您不坚持使用列表理解:
>>> mylist = ["a","b","c","d"]
>>> list(enumerate(mylist))
[(0, 'a'), (1, 'b'), (2, 'c'), (3, 'd')]
答案 3 :(得分:12)
如果您使用的是长列表,那么列表推导似乎更快,更不用说更具可读性了。
~$ 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
答案 4 :(得分:9)
这是一种方法:
>>> 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 :(得分:6)
明确关于元组。
[(i, j) for (i, j) in enumerate(mylist)]
答案 6 :(得分:0)
所有伟大的答案人。我知道这里的问题特定于枚举,但是这样的事情,只是另一个视角
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)
答案 7 :(得分:0)
如果mylist = [“ a”,“ b”,[“ x”,“ y”],“ d”]