使用 python matlib 绘制热力图
绘制世界地图
点击查看代码
import matplotlib.pyplot as plt
from mpl_toolkits.basemap import Basemap
plt.figure(figsize=(16,8))
m = Basemap()
m.drawcoastlines()
plt.show()
绘制国家轮廓
点击查看代码
m.drawcountries(linewidth=1.5)
直接定位到中国版图
点击查看代码
m = Basemap(llcrnrlon=73, llcrnrlat=18, urcrnrlon=135, urcrnrlat=53)
添加兰博投影效果
点击查看代码
m = Basemap(llcrnrlon=77, llcrnrlat=14, urcrnrlon=140, urcrnrlat=51, projection='lcc', lat_1=33, lat_2=45, lon_0=100)
绘制中国省份 需要下载中国经纬度 csv
- 可以自己下载 中国经纬度 CHN_adm_shp/CHN_adm1.zip
- 百度云链接 链接: https://pan.baidu.com/s/1vD9pYR6KVphZzR2m3ACUfw?pwd=xxoo 提取码: xxoo
点击查看代码
m.readshapefile('CHN_adm_shp/CHN_adm1', 'states', drawbounds=True)
先给中国地图上色
- gca ：Get Current Axes 获得当前图形的座标轴
- facecolor ='r' r= red
点击查看代码
from matplotlib.patches import Polygon
ax = plt.gca()
for nshape, seg in enumerate(m.states):
poly = Polygon(seg, facecolor='r')
ax.add_patch(poly)
添加人口人力热力效果
- 可以去全国第六次人口普查数据 官网下载 (http://www.stats.gov.cn/tjsj/pcsj/rkpc/6rp/indexce.htm
- 百度云链接 链接: https://pan.baidu.com/s/1En1MunNU-Ma5h5kGitY8Fw?pwd=xxoo 提取码: xxoo
增加调色板基准色
点击查看代码
cmap = plt.cm.YlOrRd
调整每个省份的颜色
点击查看代码
colors[s] = cmap(np.sqrt((pop - vmin) / (vmax - vmin)))[:3]
将颜色绘制到地图
点击查看代码
color = rgb2hex(colors[statenames[nshape]]) poly = Polygon(seg, facecolor=color, edgecolor=color)
一下完整代码
点击查看代码
import matplotlib.pyplot as plt
from mpl_toolkits.basemap import Basemap
from matplotlib.patches import Polygon
from matplotlib.colors import rgb2hex
import numpy as np
import pandas as pd
plt.figure(figsize=(16, 8))
m = Basemap(
llcrnrlon=77,
llcrnrlat=14,
urcrnrlon=140,
urcrnrlat=51,
projection='lcc',
lat_1=33,
lat_2=45,
lon_0=100
)
m.drawcountries(linewidth=1.5)
m.drawcoastlines()
m.readshapefile('CHN_adm_shp/CHN_adm1', 'states', drawbounds=True)
df = pd.read_csv('pop.csv')
df['省名'] = df.地区
df.set_index('省名', inplace=True)
colors = {}
provinces = []
for shape_dict in m.states_info:
name = shape_dict['NL_NAME_1']
p = name.split('|')
s = p[1] if len(p) > 1 else p[0]
provinces.append(s)
pop = df['人口数'][s]
colors[s] = plt.cm.YlOrRd(np.sqrt((pop - 3000000) / (100000000 - 3000000)))[:3]
ax = plt.gca()
for n_shape, seg in enumerate(m.states):
color = rgb2hex(colors[provinces[n_shape]])
poly = Polygon(seg, facecolor=color, edgecolor=color)
ax.add_patch(poly)
if __name__ == '__main__':
plt.show()
参考： https://segmentfault.com/a/1190000010871928
标签：plt,查看,python,代码,热气,点击,import,绘制,color From： https://www.cnblogs.com/guanchaoguo/p/16836311.html