python中的数据可视化——热力图 sns.heatmap()
import seaborn as sns
import matplotlib.pyplot as plt
import numpy as np
# 创建一个随机的数据表
np.random.seed(0)
df = sns.load_dataset("flights")
df = df.pivot("month", "year", "passengers")
# 使用sns.heatmap()函数绘制热力图
sns.heatmap(df, annot=True, fmt='d', cmap='YlGn')
plt.show()这段代码首先导入了seaborn和matplotlib的pyplot库，以及numpy库。然后，它创建了一个随机的数据表，并使用pivot方法重塑了数据表，以便用作热力图的输入。最后，sns.heatmap()函数用于生成热力图，其中annot=True表示显示数据注释，fmt='d'指定数字格式为整数，cmap='YlGn'指定了颜色映射方案。最后，使用plt.show()显示图形。
评论已关闭