numpy - 在numpy数组中查找包含子字符串的条目？

Question

我试图在一个数组中找到包含np.where和in条件的子字符串的条目:

import numpy as np
foo = "aa"
bar = np.array(["aaa", "aab", "aca"])
np.where(foo in bar)

这只会返回一个空数组。
为什么会这样？
有没有好的替代解决方案？

Answer 1

我们可以使用 np.core.defchararray.find 来找到foo字符串在bar的每个元素中的位置，如果找不到，它将返回-1。因此，可以通过检查foo的输出中的-1来检测每个元素中是否存在find。最后，我们将使用np.flatnonzero来获取匹配项的索引。因此，我们将有一个实现，像这样-

np.flatnonzero(np.core.defchararray.find(bar,foo)!=-1)

sample 运行-

In [91]: bar
Out[91]: 
array(['aaa', 'aab', 'aca'], 
      dtype='|S3')

In [92]: foo
Out[92]: 'aa'

In [93]: np.flatnonzero(np.core.defchararray.find(bar,foo)!=-1)
Out[93]: array([0, 1])

In [94]: bar[2] = 'jaa'

In [95]: np.flatnonzero(np.core.defchararray.find(bar,foo)!=-1)
Out[95]: array([0, 1, 2])