Numpy相关混淆[重复]

Question

我想计算 2 个数组之间的相关性。为此，我想使用 NumPy。

我在一个小例子中使用了numpy.correlate 函数：

import numpy as np

a = [1, 2, 3]
b = [2, 3, 4]

np.correlate(a, b)
>>> np.array([20])

我真的不知道如何解释这个结果。我想要的是一个介于 -1 和 1 之间的数字来表示相关性，其中 1 表示数组正相关，-1 表示数组负相关。

我怎样才能得到这个号码？

Answer 1

您使用了错误的功能。您正在寻找numpy.corrcoef，它实际上计算相关系数。

a = [1, 2, 3]
b = [2, 3, 4]

>>> np.corrcoef(a, b)
array([[ 1.,  1.],
       [ 1.,  1.]])

As mentioned by Hooked，这会从协方差矩阵中返回一个值矩阵。

如果你想要 Pearson 相关系数，你可以使用 pearsonr from scipy.stats.stats。 Hooked's answer here 是此方法的正确实现。