如何查找列表中所有唯一元素的所有索引

Question

给出以下列表：

list_ex = ['s1', 's2', 's1', 's4', 's2', 's3', 's1']

如何找到所有不同元素的所有索引？

例如，对于s1，这将是locations [0, 2, 6]。

我认为我可以通过在不同的元素上运行循环来做到这一点list(set(list_ex))，然后执行 anp.where来查找位置？

Answer 1

我发现这pandas似乎针对此类问题进行了优化。import randomimport pandas as pdx = [f's{i}' for i in range(1000)]l = [random.choice(x) for _ in range(2000000)]output = pd.DataFrame(l).groupby([0]).indices它可以比enumerate最佳场景（组的大小较大）快 3 倍，而在组的大小较小（每组 1 - 2 个项目）的情况下慢 3 倍。

Answer 2

您可以循环遍历元素，构建一个字典，将元素映射到该元素的索引列表。使用defaultdictof 类型list对此很方便，因为第一次读取新元素时会自动获得一个空列表。from collections import defaultdictlist_ex = ['s1', 's2', 's1', 's4', 's2', 's3', 's1']indices = defaultdict(list)for i, v in enumerate(list_ex):&nbsp; indices[v].append(i)print(indices)这将打印以下内容：defaultdict(<class 'list'>, {'s1': [0, 2, 6], 's2': [1, 4], 's4': [3], 's3': [5]})

Answer 3

这是使用列表理解的简短解决方案：locations = [el[0] for el in enumerate(list_ex) if el[1] == "s1"]解释枚举创建位置/元素对的列表，它看起来像这样：[(0, 's1'), (1, 's2'), (2, 's1'), (3, 's4'), (4, 's2'), (5, 's3'), (6, 's1')]下面的代码得到相同的结果，只是以 for 循环形式显示：target = 's1'locations = []for el in enumerate(list_ex):&nbsp; &nbsp; if el[1] == target:&nbsp; &nbsp; &nbsp; &nbsp; locations.append(el[0])