import pandas as pd
data = pd.read_excel('./dealing_data.xlsx')
import seaborn as sns
import numpy as np
import matplotlib.pyplot as plt
f, ax= plt.subplots(figsize = (14, 10))
corr = data.corr()
# print(corr)
sns.heatmap(corr,cmap='RdBu', linewidths = 0.05, ax = ax, annot=True)
# 设置Axes的标题
ax.set_title('Correlation between features')
plt.show()
plt.close()
f.savefig('person_result.jpg', dpi=100, bbox_inches='tight')
其中heatmap()方法中有annot参数,默认为False,不显示每个颜色的数字,如果设置为:annot=True, 则在每个热力图上显示数字。效果如下: