2

我读了这篇文章:如何在列表中找到所有出现的元素? 如何在列表中查找所有出现的元素?

给出的答案是:

indices = [i for i, x in enumerate(my_list) if x == "whatever"]

我知道这是列表理解,但我无法分解这段代码并理解它。有人可以给我一块吃吗?


如果执行以下代码:我知道 enumerate 只会创建一个元组:

l=['a','b','c','d']
enumerate(l)

输出:

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

如果有更简单的方法,我也会对此持开放态度。

4

2 回答 2

7

indices = [i for i, x in enumerate(my_list) if x == "whatever"]相当于:

# Create an empty list
indices = []
# Step through your target list, pulling out the tuples you mention above
for index, value in enumerate(my_list):
    # If the current value matches something, append the index to the list
    if value == 'whatever':
        indices.append(index)

结果列表包含每个匹配项的索引位置。采用相同的for结构,您实际上可以更深入地遍历列表列表,从而使您陷入 Inception 式的疯狂螺旋:

In [1]: my_list = [['one', 'two'], ['three', 'four', 'two']]

In [2]: l = [item for inner_list in my_list for item in inner_list if item == 'two']

In [3]: l
Out[3]: ['two', 'two']

这相当于:

l = []
for inner_list in my_list:
  for item in inner_list:
    if item == 'two':
      l.append(item)

您在开头包含的列表理解是我能想到的最 Pythonic 的方式来完成您想要的。

于 2012-11-20T17:48:34.277 回答
0
indices = []
for idx, elem in enumerate(my_list):
    if elem=='whatever':
        indices.append(idx)
于 2012-11-20T17:47:44.487 回答