用seaborn作聚类热图

代码中使用的文件下载地址：
https://codechina.csdn.net/mirrors/mwaskom/seaborn-data?utm_source=csdn_github_accelerator
其中fs.csv为flights.csv文件

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


sns.set()

data = pd.read_csv("fs.csv")
#print(data.head())

flights = data.pivot("month","year","passengers")
## 绘制x-y-z的热力图，比如 年-月-销量 的聚类热图
g = sns.clustermap(flights, fmt='d', cmap='YlGnBu')
ax = g.ax_heatmap
label_y = ax.get_yticklabels()
plt.setp(label_y,rotation=360, horizontalalignment='left')
plt.show()

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


sns.set()

data = pd.read_csv("iris.csv")
#print(data.head())

species = data.pop("species")
#设置图片大小
g= sns.clustermap(data, fmt="d",cmap='YlGnBu', figsize=(6, 9))
ax = g.ax_heatmap
label_y = ax.get_yticklabels()
plt.setp(label_y, rotation=360, horizontalalignment='left')
#设置图片名称，分辨率，并保存
#plt.savefig('cluster.tif', dpi=300)
plt.show()