我有一个字符串“列表”,我需要获取列表中等于模式的所有元素的索引数组。
我不需要使用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。
最佳答案
这是一个 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')
关于python - 字符串列表的索引数组,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/49026751/