代码中使用的文件下载地址:
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()