numpy 中判断某字符串 array 是否含有子字符串
网上找了一大圈没找到, 还有好多错的...
numpy.char.count(a, sub, start=0, end=None)
该函数是用来计数 sub 在 a 中出现多少次, 我们稍加修改就能达到我们要的效果.
numpy.char.count(a, sub, start=0, end=None) != 0
函数的具体介绍看官方文档.
举例:
import numpy as np
a = np.array(['abc', 'akjs', '23ha', 'sdfg', 'arge', '8908d', 'aaaab', 'hhdaa', 'wer'], dtype='str')
np.char.count(a, 'a')
# array([1, 1, 1, 0, 1, 0, 4, 2, 0])
np.char.count(a, 'a') != 0
# array([ True, True, True, False, True, False, True, True, False])