本想着有什么函数可以直接调用的, 结果网上找了一大圈没找到, 还有好多错的…

1
numpy.char.count(a, sub, start=0, end=None)

该函数是用来计数 sub 在 a 中出现多少次, 我们稍加修改就能达到我们要的效果.

1
numpy.char.count(a, sub, start=0, end=None) != 0

函数的具体介绍看官方文档.

举例:

1
2
3
4
5
6
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])