我想在包含子字符串的数组中找到所有索引,在本例中为 substring = "contig_"
。我开始逐行迭代,然后迭代数组中的每个元素,但这是它获得的最强大的力量。
在 numpy 或 scipy 中是否有任何函数可以比蛮力方法更快?
A = np.array([['K00180:55:H3NHMBBXX:7:1101:30340:1068', '83',
'contig_1758_2278_4341_-', '1487', '60', '140M', '=', '1334',
'293', "=",
'*', 'RG:Z:RG_0', 'MD:Z:23A30A85', 'NM:i:2\n'],
['K00180:55:H3NHMBBXX:7:1101:30340:1068', '163',
'contig_1758_2278_4341_-', '1334', '60', '87M1I3M1D17M', '=',
'1487', '293', "contig_1297_3232_198298_+",
'*', 'RG:Z:RG_0', 'MD:Z:31G3G2G6T6C6A9C4T15^G17', 'NM:i:10\n'],
['K00180:55:H3NHMBBXX:7:1101:28026:1103', '83',
'contig_1281_415_1704_-', '514', '60', '142M', '=', '396', '260', "=",
'*', 'RG:Z:RG_0', 'MD:Z:11C130', 'NM:i:1\n']],
dtype='<U149')
for row in A:
print(np.where(["contig_" in x for x in row])[0])
# [2]
# [2 9]
# [2]
最佳答案
np.char
是一组函数,它们将字符串方法应用于像您这样的数组的元素。所以使用 find
函数:
In [311]: np.char.find(A, 'contig')
Out[311]:
array([[-1, -1, 0, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1],
[-1, -1, 0, -1, -1, -1, -1, -1, -1, 0, -1, -1, -1, -1],
[-1, -1, 0, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1]])
-1 代表未找到的元素,0 或更大代表找到。
In [312]: np.where(np.char.find(A, 'contig')>=0)
Out[312]: (array([0, 1, 1, 2], dtype=int32), array([2, 2, 9, 2], dtype=int32))
In [313]: A[_]
Out[313]:
array(['contig_1758_2278_4341_-', 'contig_1758_2278_4341_-',
'contig_1297_3232_198298_+', 'contig_1281_415_1704_-'],
dtype='<U149')
像这样的函数必须遍历元素,并应用相应的字符串方法,因此它们不像通常的 numpy 数字代码那么快,但它们比您自己的迭代要容易得多。
np.vectorize
或 np.frompyfunc
也可用于将函数应用于数组的每个元素。它们也进行迭代,因此与您自己的迭代相比并没有显着的加速。我仍然发现 frompyfunc
通常提供 30% 的加速。
In [331]: f=np.frompyfunc(lambda x: x.find('contig'), 1,1) # like char.find
In [332]: f=np.frompyfunc(lambda x: 'contig' in x, 1,1) # your 'in'
In [333]: f(A)
Out[333]:
array([[False, False, True, False, False, False, False, False, False,
False, False, False, False, False],
[False, False, True, False, False, False, False, False, False, True,
False, False, False, False],
[False, False, True, False, False, False, False, False, False,
False, False, False, False, False]], dtype=object)
In [334]: np.where(f(A))
Out[334]: (array([0, 1, 1, 2], dtype=int32), array([2, 2, 9, 2], dtype=int32))
关于python - 查找 NumPy 数组中包含子字符串的所有位置(最有效?),我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/42148264/