numpy.corrcoef 计算相关系数

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()