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]