我曾在其他地方看到过相同的主题,但对我的问题没有真正的答案。我有一个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的元素明智比较的数组。

09-11 18:42
查看更多