用Python和Numpy计算协方差

Question

我试图弄清楚如何使用PythonNumpy函数cov计算协方差。当我将其传递给两个一维数组时，我得到了一个2x2的结果矩阵。我不知道该怎么办。我不太擅长统计，但我相信在这种情况下的协方差应该是一个整数。
这就是我想要的。我写了我自己的：

def cov(a, b):

    if len(a) != len(b):
        return

    a_mean = np.mean(a)
    b_mean = np.mean(b)

    sum = 0

    for i in range(0, len(a)):
        sum += ((a[i] - a_mean) * (b[i] - b_mean))

    return sum/(len(a)-1)

可以，但是我认为Numpy版本要有效得多，如果我能弄清楚如何使用它的话。

有人知道如何使Numpy cov函数像我写的那样执行吗？

谢谢，

Answer 1

当a和b是一维序列时，numpy.cov(a,b)[0][1]等效于cov(a,b)。

返回的2x2数组的np.cov(a,b)元素等于

cov(a,a)  cov(a,b)

cov(a,b)  cov(b,b)

（再次cov是您在上面定义的功能。）