python - 字符串列表的索引数组

Question

我有一个字符串“列表”，我需要获取列表中等于模式的所有元素的索引数组。

我不需要使用Python列表，我可以使用另一个数据结构，但我不知道是哪个。

示例:

my_list = ['foo', 'bar', 'hello', 'foo', 'goodbye']
pattern = 'foo'

我想要的结果是这样的:

my_mask = [True, False, False, True, False]

所以我可以用这个掩码索引一个 numpy 数组:

selected_items = my_array[my_mask]

但是，列出 list 并执行以下操作:

my_list == pattern

不返回掩码，仅返回 False。

Answer 1

这是一个 numpy 解决方案:

import numpy as np

my_list = np.array(['foo', 'bar', 'hello', 'foo', 'goodbye'])
pattern = 'foo'

mask = my_list == pattern
# array([ True, False, False,  True, False], dtype=bool)

my_list[mask]
# array(['foo', 'foo'], 
#       dtype='<U7')