比较两个字符串列表并获取重复索引

Question

我想得到重复的值（字符串）的索引。 例如：

a=['iii','jjj','rrr']
b=['iii','lll','yyy','ttt','jjj']    
s=numpy.where(a==b)

我希望s返回[0,4]，但此刻它只返回[0]，因为它们是相同的值并且在列表中的位置相同。

Answer 1

numpy.where使用numpy.in1d：

>>> np.where(np.in1d(b, a))[0]
array([0, 4]

Answer 2

您可以在列表理解中使用max和min功能：

>>> [i for i,j in enumerate(max(a,b,key=len)) if j in min(a,b,key=len)]
[0, 4]