我曾在其他地方看到过相同的主题,但对我的问题没有真正的答案。我有一个numpy数组,我需要找到一个数字的索引。
a=np.argsort(cosine_similarity(tfidf_matrix[11:12], tfidf_matrix)) #numbers are from 0 to 11
b=np.equal(a,10)
# b values are [[False False False False False False False False True False False False]]
如何获取返回索引8的信息? (数组中真实值的索引)
最佳答案
您正在寻找numpy.where
b = np.where(a==10)
这里的
b
是一个数组,其中包含与您的条件相匹配的项的索引。如果只对第一个元素感兴趣,可以选择第一个元素(b[0]
)。numpy.equal上的文档说:
按元素返回(x1 == x2)。
这正是您收到的,一个包含数组
a
和值10
的元素明智比较的数组。