Python相关系数矩阵热力图(二)

上面图片通过下面代码实现

同时 df 内部的数据是：

AB

0	0.180270	0.019475
1	0.463219	0.724934
2	0.420204	0.485427

由于我设置了随机数种子，各位的数据也会是跟我一样的~
图片当然也是类似的啦~

datafram.corr() 会根据不同列的元素来进行运算，算出各个列之间的相关系数矩阵。（各个列！）

import seaborn as sns import matplotlib.pyplot as plt import pandas as pd import numpy as npdef draw(df):dfData = df.corr()plt.subplots(figsize=(4, 4)) # 设置画面大小sns.heatmap(dfData, annot=True, vmax=1, square=True, cmap="Blues")plt.savefig('./BluesStateRelation.png')plt.show()if __name__ == '__main__':prng = np.random.RandomState(11)s = prng.rand(3, 2)df = pd.DataFrame(s, columns=["A", "B"])print(df)draw(df)

总结

以上是生活随笔为你收集整理的Python相关系数矩阵热力图(二)的全部内容，希望文章能够帮你解决所遇到的问题。

如果觉得生活随笔网站内容还不错，欢迎将生活随笔推荐给好友。