基于Python绘制矩阵热图

数据表格式
基于Python绘制矩阵热图_第1张图片
代码如下:

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

# 加载数据表
data = pd.read_csv("DLTCQ2000.csv",header=0, index_col=0)
re_data = pd.DataFrame(data)
rc = {'font.sans-serif': 'SimHei',
      'axes.unicode_minus': False}
sns.set(context='notebook', style='ticks', rc=rc,font_scale=1.2)
f, ax = plt.subplots(figsize=(16, 10))  # 设置图大小
sns.heatmap(
      data,
      annot=True, #是否显示数据
      fmt='.3f', # 设置数值显示的小数位数
      annot_kws={'font':'Times New Roman','size':13},  # 设置字体和大小
      cmap='RdBu_r', # 设置配色
      square=1,  # 是否每个方格都是正方形
      cbar=True, # 是否绘制颜色条
      ax=ax,
      center=0.35 # 设置色彩居中值)
)
plt.savefig("矩阵热图.png",dpi=600)
plt.show()

效果图:

你可能感兴趣的:(python,矩阵,开发语言)