Numpy语法的意思

时间:2013-04-01 11:42:56

标签: python numpy

在使用python / numpy时,我遇到了以下语法:

np.mean(predicted == docs_test.target)

其中参数的类型为numpy.ndarray

这里==的含义是什么?

谢谢, Jaideep

3 个答案:

答案 0 :(得分:3)

假设predicteddocs_test.target是两个大小相同的数组,这将计算两个数组完全一致的元素分数。

例如,

In [1]: import numpy as np

In [2]: a = np.array([1, 2, 3, 4, 5, 6, 7])

In [3]: b = np.array([1, 3, 7, 4, 5, 0, -10])

In [4]: np.mean(a == b)
Out[4]: 0.42857142857142855

这告诉我们,约有43%的案例(7个案例中有3个),ab达成一致。

答案 1 :(得分:2)

如果predicteddocs_test.target都是numpy数组,那么==将返回一个新数组,1代替匹配的元素,0元素不同。该数组的mean将为您提供相似度,基本上为numberofmatchingelements / lengthofpredictedarray

答案 2 :(得分:1)

来自docs

  

每个......比较(==<><=>=!=)是   相当于相应的通用函数

在这种情况下,相应的通用功能是numpy.equal

  

numpy.equal(x1, x2[, out])

     

以元素为单位返回(x1 == x2