使用numpy计算皮尔逊相关系数
时间: 2023-05-31 07:05:47 浏览: 327
python计算皮尔逊系数
可以使用numpy中的corrcoef函数来计算皮尔逊相关系数。该函数接受两个数组作为输入,返回一个矩阵,其中对角线上的元素为1(因为每个变量与自身的相关系数为1),其他元素为两个变量之间的相关系数。
例如,计算两个数组a和b之间的相关系数,可以使用以下代码:
```python
import numpy as np
a = np.array([1, 2, 3, 4, 5])
b = np.array([5, 4, 3, 2, 1])
correlation_matrix = np.corrcoef(a, b)
print(correlation_matrix)
```
输出结果为:
```
[[ 1. -1.]
[-1. 1.]]
```
可以看到,a和b之间的相关系数为-1,表示它们之间存在强烈的负相关关系。
阅读全文