Python枚举使用列表理解和如果语句

Question

列举理解枚举只有当if语句为True时才有枚举的方法吗？

例如，下面的代码枚举if语句是True还是False，我想有一个连续的索引。

my_list = [0,1,2] [i for i, w in enumerate(my_list) if w != 1]

我也试过以下，但你不能使用通在列表解析里像下面。

[i if w != 1 else pass for i, w in enumerate(my_list)] 

Answer 1

有。一个简单的方法是使用itertools.count：

>>> from itertools import count 
>>> c = count(0) 
>>> my_list = [0,1,2] 
>>> [next(c) for w in my_list if w != 1] 
[0, 1] 

但是：这涉及到一个列表的理解，这仅仅是......不好内部状态变化。请注意，如果您再次运行您的理解，这将有不幸的效果：

>>> [next(c) for w in my_list if w != 1] 
[2, 3] 
>>> [next(c) for w in my_list if w != 1] 
[4, 5] 

相反，我会做这样的事情：

>>> range(sum(w != 1 for w in my_list)) 
range(0, 2) 

Answer 2

如何把一个列表理解（发电机）列表解析里？

my_list = [0,1,2] 
[i for i, w in enumerate((_ for _ in my_list if _ != 1)) ] 

输出：

[0, 1]