皮尔逊相关系数分析

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

plt.rcParams['font.family'] = ['sans-serif']
plt.rcParams['font.sans-serif'] = ['SimHei']
plt.rcParams['axes.unicode_minus']=False

# 生成随机数据集 数据见同名微信公众号

data = pd.read_excel('data.xlsx').values
print(data.shape)
df = pd.DataFrame(data, columns=['变量x', '2*x', '随机数', '3*x','随机数'])

# 计算相关性矩阵
corr_matrix = df.corr()

# 绘制相关性矩阵
plt.figure(figsize=(8, 6))
sns.heatmap(corr_matrix, annot=True, cmap='Blues', vmin=-1, vmax=1)
## coolwarm,Blues，viridis
plt.title('Correlation Matrix')
plt.show()