numpy.corrcoef 计算相关系数

最新推荐文章于 2025-10-10 16:19:51 发布

原创最新推荐文章于 2025-10-10 16:19:51 发布 · 3.7k 阅读

7 ·

CC 4.0 BY-SA版权

文章标签：

#numpy #python #算法 #数据分析

numpy.corrcoef(x, y=None, rowvar=True)

x：(array_like)，rowvar=True时，行特征，列记录。rowvar=False相反
y：(array_like，可选)，一组额外的特征和值，数组形状与x相同

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt

beer = pd.read_csv("./data/beer_data.txt", sep=' ')
X = beer[['calories', 'sodium', 'alcohol', 'cost']]

cov = np.corrcoef(X.T)
img = plt.matshow(cov, cmap=plt.cm.winter)
plt.colorbar(img, ticks=[cov.min(), 0, 1])
plt.xticks(np.arange(len(X.keys())), X.keys())
plt.yticks(np.arange(len(X.keys())), X.keys())
plt.show()