Python Matplotlib画一个热力图

最新推荐文章于 2024-05-06 00:08:33 发布

LawLietW

最新推荐文章于 2024-05-06 00:08:33 发布

阅读量759

点赞数

文章标签： python matplotlib 开发语言

本文链接：https://blog.csdn.net/LawLietW/article/details/125960465

版权


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

## Import 


df = pd.read_excel("text4.xlsx", sheet_name = "before_2019")
df2 = pd.read_excel("text4.xlsx", sheet_name = "after_2019")

# Read Data

d = df.corr()
d2 = df2.corr()

# Corr matrix

# Plot!
plt.figure(figsize= (10, 16))
plt.subplot(2,1,1)
sns.heatmap(d2,annot = True,vmax = 1,square = True,cmap = "Reds")
plt.xticks(rotation=38)
plt.title("National Cooperation Before 2019")
plt.subplot(2,1,2)
sns.heatmap(d,annot = True,vmax = 1,square = True,cmap = "Reds")
plt.xticks(rotation=38)
plt.title("National Cooperation After 2019")
plt.show()

导入的数据大概是这样的excel：
请添加图片描述

结果如下：
在这里插入图片描述