python相关性分析及可视化

数据准备

需要一个类似excel文件的数据。进行每列字段之间的相关性分析。

计算

导入第三方库。

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

进行计算

f_path = r'相关性.xlsx'
df = pd.read_excel(f_path)
corr = df.corr()
print(corr)

可视化

heatmap = sns.heatmap(corr)
plt.show()

如果发现中文字体有问题，只需要在最前面加上下面两句。

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