具有多个元素的数组的真值是不明确的。使用a.any()或a.all()-Numpy库

我在其他地方也看到过相同的话题,但是我的问题没有真正的答案。我有一个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的信息? (数组中真实值的索引)

q5858255 回答:具有多个元素的数组的真值是不明确的。使用a.any()或a.all()-Numpy库

您正在寻找numpy.where

b = np.where(a==10)

这里b将是一个数组,其中包含与您的条件匹配的项目的索引。如果只对第一个元素感兴趣,可以选择第一个元素(b[0]

numpy.equal上的文档说:

  

按元素返回(x1 == x2)。

这正是您收到的,一个包含数组a和值10的逐元素比较的数组。

本文链接:https://www.f2er.com/3156203.html

大家都在问