seaborn热图绘制与参数

seaborn热图绘制与参数

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

data = pd.read_csv('./dist.csv', index_col=0) #传入的数据可以是DataFrame、array
print(data.shape)

#pd.set_option('max_colwidth',150)
#cmap = sns.cm.rocket_r
sns.set(font_scale=0.4) #修改字体大小
#plt.figure(figsize=(15,15))
plt.figure(dpi= 200)
sns.heatmap(data, cmap="PiYG").invert_yaxis() #invert_yaxis()翻转y轴 cmap="PiYG"改变热图颜色
#plt.yticks(rotation=0)
plt.savefig('./figure/heatmap.jpg') #bbox_inches='tight'
plt.show()