如何在python数组中查找字符串的一部分

Question

我有一个数组，有一列数字和一列字符串。我需要找到包含字符串某一部分的索引。示例：

array([['4', 'red crayon'],
       ['2', 'purple people eater'],
       ['6', 'red hammer'],
       ['12', 'green screwdriver']])

我在入门编程，所以我的导师告诉我在数组[index]中使用if'stringpart'来获得一个布尔数组，用于解决问题。我认为应该发生什么：

input:
'red' in array[:,1]
output:
(true,false,true,false)

事实上，我完全错了。我在函数之外测试了代码，并将其应用到只包含字符串的数组中，结果仍然是false。但我用一支完整的“红色蜡笔”试了一下，结果是真的。这个部分字符串代码适用于教师和其他学生。为什么不适合我？你知道吗

Answer 1

对于矢量化解决方案，可以使用np.char模块：

a = np.array([['4', 'red crayon'],
              ['2', 'purple people eater'],
              ['6', 'red hammer'],
              ['12', 'green screwdriver']])
np.char.find(a[:, 1], 'red') > -1
# array([ True, False,  True, False], dtype=bool)