我有一个长度为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 string?
编辑:
我想在A
中找到'zzz'外观的索引答案 0 :(得分:6)
不需要对此有所了解,您可以通过列表理解和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
运算符也适用于常规字符串,所以它没有实际意义。
答案 1 :(得分:2)
这里的问题是你的字符串数组的性质。
如果我将数组设为:
In [362]: x=np.array(['one','two','three'])
In [363]: x
Out[363]:
array(['one', 'two', 'three'],
dtype='<U5')
In [364]: type(x[0])
Out[364]: numpy.str_
元素是特殊类型的字符串,隐式填充为5个字符(最长的&#39; np.char方法适用于此类数组
In [365]: np.char.find(x,'one')
Out[365]: array([ 0, -1, -1])
但是如果我创建一个包含字符串的对象数组,它会产生错误
In [366]: y=np.array(['one','two','three'],dtype=object)
In [367]: y
Out[367]: array(['one', 'two', 'three'], dtype=object)
In [368]: type(y[0])
Out[368]: str
In [369]: np.char.find(y,'one')
...
/usr/lib/python3/dist-packages/numpy/core/defchararray.py in find(a, sub, start, end)
...
TypeError: string operation on non-string array
通常情况下,必须将对象数组视为列表。
In [370]: y
Out[370]: array(['one', 'two', 'three'], dtype=object)
In [371]: [i.find('one') for i in y]
Out[371]: [0, -1, -1]
In [372]: np.array([i.find('one') for i in y])
Out[372]: array([ 0, -1, -1])
np.char
方法很方便,但速度不快。他们仍然必须遍历数组,对每个元素应用常规字符串操作。
答案 2 :(得分:1)
你可以尝试这个:
np.core.defchararray.find(A.astype(str), 'zzz')