画一个简单的热力图
import matplotlib.pyplot as plt
import numpy as np
np.random.seed(0)
import seaborn as sns
# 初始化参数
sns.set()
uniform_data = np.random.rand(3, 3)
heatmap = sns.heatmap(uniform_data)
plt.show()
效果图
利用自己的数据,加载数据集
import seaborn as sns
import matplotlib.pyplot as plt
#这里的第一个位置为文件名,即example.csv,data_home为本地数据集的路径
data = sns.load_dataset('example', data_home='C:/Users/1\PycharmProjects/')
# 取出这三个属性画热力图,坐标点的位置是passengers
data = data.pivot_table('ALB', 'Lat', 'Lon')
ax = sns.heatmap(data, cmap='Greens').invert_yaxis()
plt.show()
效果图如下:
为了得到坐标轴从小到大,可以使用**.invert_yaxis()**函数实现y坐标轴的翻转
效果为下
ax = sns.heatmap(data, cmap='Greens').invert_yaxis()
结果为
如果想调整颜色,可以根据下面的博客进行调整。
博客:https://blog.csdn.net/weixin_39777626/article/details/95892284