Answers for "numpy.where() for substring"

0

numpy.where() for substring

foo = 'aa'
bar = np.array(['aaa', 'aab', 'aca'])
out = [i for i, v in enumerate(bar) if foo in v]
# out = [0, 1]

bar = ['aca', 'bba', 'baa', 'aaf', 'ccc']
out = [i for i, v in enumerate(bar) if foo in v]
# out = [2, 3]
Posted by: Guest on January-18-2022

Python Answers by Framework

Browse Popular Code Answers by Language