我有一个包含一列数字和一列字符串的数组。我需要找到包含字符串特定部分的索引。例子:
array([['4', 'red crayon'],
['2', 'purple people eater'],
['6', 'red hammer'],
['12', 'green screwdriver']])
我在入门编程,所以我的导师告诉我在 array[index] 中使用 if 'stringpart' 来获得一个布尔数组,用于我们的问题解决。我认为应该发生的事情:
input:
'red' in array[:,1]
output:
(true,false,true,false)
实际发生的事情是我全都是假的。我在函数之外测试了代码,并将其应用于只有字符串的数组,但它仍然给我错误。但是我用一根完整的“红色蜡笔”试了一下,结果是真的。此部分字符串代码适用于教师和其他学生。为什么它不适合我?
对于矢量化解决方案,您可以使用该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)
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句