在列表理解中使用枚举的Python
假设我有这样一个列表:
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
所以,我的问题是:在列表理解中使用枚举的正确方法是什么?
尝试这个:
[(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')]
为了清楚起见,这与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')])
正如ÓscarLópez所说,你可以直接传递枚举元组:
>>> [t for t in enumerate('abcdef') ] [(0, 'a'), (1, 'b'), (2, 'c'), (3, 'd'), (4, 'e'), (5, 'f')]
或者,如果你不坚持使用列表理解:
>>> mylist = ["a","b","c","d"] >>> list(enumerate(mylist)) [(0, 'a'), (1, 'b'), (2, 'c'), (3, 'd')]
如果你使用长列表,看起来列表的理解速度更快,更不用提更多的可读性。
~$ 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
这是一个办法:
>>> 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)]
所有伟大的答案家伙。 我知道这里的问题是枚举的具体问题,但是对于这样的事情,只是另一个angular度而已
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)