python 在 numpy 字符串数组中查找字符串模式

Question

我有一个长度为 100 的字符串 'A' 的 numpy 数组，它们是不同大小的句子。它是字符串而不是 numpy 字符串

>>> type(A[0])
<type 'str'>

我想在 A 中找到字符串的位置，其中包含某些模式，如 'zzz'。

我试过了

np.core.defchararray.find(A, 'zzz')

给出错误:

TypeError: string operation on non-string array

我假设我需要将 A 中的每个“str”更改为 numpy 字符串？

编辑:

我想在A中找到'zzz'出现的索引

Answer 1

不必花太多心思，您可以通过列表理解和 in 运算符获得索引列表:

>>> import numpy as np
>>> lst = ["aaa","aazzz","zzz"]
>>> n = np.array(lst)
>>> [i for i,item in enumerate(n) if "zzz" in item]
[1, 2]

请注意，此处数组的元素实际上是 numpy 字符串，但 in 运算符也适用于常规字符串，因此没有实际意义。