在numpy数组中查找单词

Question

import re
a=input('Enter your stuff: ')
s=re.sub('[a-zA-Z^]','',a)
print([int('-'+i[0]) if s[s.index(i)-2]=='-' else int(i[0]) for i in re.split(' [+|-] ',s)])

在英语词典中，dictionary变量是一个具有成千上万个单词的numpy数组，当查找单词的索引时，numpy还会返回所有包含word2作为子字符串的字符串的索引。

有人知道如何解决此问题吗？我只想返回完美的字符串匹配

谢谢

Answer 1

a = np.array(['apple', 'orange', 'apple', 'banana'])

arr_index = np.where(a == 'apple')

print arr_index

print a[arr_index]

请参阅此post