对其中的参数进行解释
plt.subplots(figsize=(9, 9))
设置画面大小,会使得整个画面等比例放大的- sns.heapmap()这个当然是用来生成热力图的啦
- df是DataFrame, pandas的这个类还是很常用的啦~
- df.corr()就是得到这个dataframe的相关系数矩阵
- 把这个矩阵直接丢给sns.heapmap中做参数就好啦
- sns.heapmap中
annot=True
,意思是显式热力图上的数值大小。 - sns.heapmap中
square=True
,意思是将图变成一个正方形,默认是一个矩形 - sns.heapmap中
cmap="Blues"
是一种模式,就是图颜色配置方案啦,我很喜欢这一款的。 - sns.heapmap中
vmax
是显示最大值
import seaborn as sns
import matplotlib.pyplot as plt
def test(df):
dfData = df.corr()
plt.subplots(figsize=(9, 9))
sns.heatmap(dfData, annot=True, vmax=1, square=True, cmap="Blues")
plt.savefig('./BluesStateRelation.png')
plt.show()