【matplotlib基础】--结合地图
如果分析的数据与地域相关,那么,把分析结果结合地图一起展示的话,会让可视化的效果得到极大的提升。
比如,分析各省GDP数据,人口数据,用柱状图,饼图之类的虽然都可以展示分析结果,
不过,如果能在全国的地图上展示各省的分析结果的话,会让人留下更加深刻的印象。
将数据的分析结果展示在地图上,难点在于:
- 如何绘制地图,地图是展示数据的基础,如何绘制出需要的地图区域是第一步
- 数据和地图关联,数据最终要显示在地图上,数据如何与地理坐标关联也是重要的一步
- 地图上展示数据,也就是要在地图上绘制不同的颜色或者几何形状来表达不同的数据
解决了上面3个难题,就能够结合地图做一些基本的数据展示了。
本篇通过一个基于南京各个区地理信息的分析示例,来演示如何一步步通过地图来展示分析结果。
1. 绘制地图
第一步是绘制地图,其实地图就是一块块不规则的多边形拼接起来的。
在本篇的示例中,各个多边形就是南京的各个区。
绘制多边形不难,难的是如何得到各个多边形的顶点坐标。
好在现在有很多的开放地理信息平台,可以让我们获取到想要的地理信息。
1.1. 获取地理信息
比如,通过阿里的DataV数据可视化平台,可以获取南京各个区的地理范围信息。
在这个平台上,左边选择区域,右边会生成对应范围的地理信息的数据。
地理信息数据是json
格式。
1.2. 展示地理信息
为了读取地理信息数据并展示,需要用到一个 GeoPandas
的库。
用 pip
安装很简单:
$ pip install geopandas
geopandas
可以直接读取DataV数据可视化平台生成的JSON数据并展示。
import geopandas as gpd
df_geo = gpd.read_file(
"https://geo.datav.aliyun.com/areas_v3/bound/320100_full.json"
)
df_geo
json
文件的HTTP
地址DataV数据可视化平台就是在上生成的。
这个文件中的关键字段就是 geometry,其中的内容就是各个区的多边形形状的各个顶点的经纬度坐标。
展示数据:
import matplotlib.pyplot as plt
fig, ax = plt.subplots()
fig.set_size_inches(8, 8)
# df_geo就是上面通过geopandas读取的数据
df_geo.plot(
ax=ax,
column="name",
cmap="plasma",
edgecolor="k",
legend=True,
legend_kwds={"loc": "lower left"}
)
plt.show()
其中主要参数的含义:
ax
:matplotlib
生成的子图,这里就是数据要绘制的位置column
:作为不同区域的名称,这里就是南京各个区的名称cmap
:一组颜色,分别对应不同的区edgecolor
:每个多边形边缘的颜色,这里设置的是黑色legend
:是否显示图例legend_kwds
:设置图例的配置信息,这里只设置了图例的位置
2. 地图和数据关联
首先生成一些测试数据:
# 删除 df_geo 中一些不必要的列
df_geo = df_geo.drop(columns=["childrenNum", "level", "parent", "subFeatureIndex"])
# 南京各个区的名称
area_names = df_geo.loc[:, "name"]
# df_val为测试数据,其中name列是各个区的名称
# value 列是一些随机数,模拟各个区的人口,GDP等数据
df_val = pd.DataFrame({
"name": area_names.tolist(),
"value": np.random.randint(10, 10000, len(area_names))
})
测试数据 df_val
模拟其他途径得到的业务数据,
下面要将 df_val
中的指展示到地图上,就要先把 df_val
和 df_geo
结合起来。
# 以 name 列作为结合两个数据集的依据
df = pd.merge(df_geo, df_val, on="name", how="left")
df
这样,我们就得到了同时包含地理信息(geometry)和业务数据(value)的数据集了。
3. 地图上展示数据
接下来就是展示数据了,下面演示两种在地图上展示数据的方式。
3.1. 热力图方式
根据 value 值的不同,用渐变色来显示不同区的颜色。
fig, ax = plt.subplots()
fig.set_size_inches(8, 8)
ax.axis("off")
df.plot(
ax=ax,
column="value",
cmap="plasma",
edgecolor="k",
legend=True,
legend_kwds={'label': "value", 'shrink':0.5},
)
for index in df.index:
x = df.iloc[index].geometry.centroid.x
y = df.iloc[index].geometry.centroid.y
name = df.iloc[index]["name"]
if name in ["建邺区", "鼓楼区", "玄武区", "秦淮区"]:
ax.text(x, y, name, ha="center", va="center", fontsize=8)
else:
ax.text(x, y, name, ha="center", va="center")
plt.show()
其中,"建邺区", "鼓楼区", "玄武区", "秦淮区" 四个区的面积比较小,
所以字体稍微调小了一些。
3.2. 分类展示
把 value
值分为3类,每类用不同的形式来表示。
- value < 3000
- 3000 <= value < 5000
- value >= 5000
import matplotlib.patches as mpatches
fig, ax = plt.subplots()
fig.set_size_inches(5, 10)
ax.axis("off")
legend_list = []
df[df["value"] < 3000].plot(
ax=ax,
color="lightblue",
edgecolor="k",
hatch="\\\\",
legend=False,
)
legend_list.append(
mpatches.Patch(
facecolor="lightblue",
edgecolor="black", hatch="\\\\", label="value<3000"
)
)
df[(df["value"] >= 3000) & (df["value"] < 5000)].plot(
ax=ax,
color="lightgreen",
edgecolor="k",
hatch="o",
legend=False,
)
legend_list.append(
mpatches.Patch(
facecolor="lightgreen",
edgecolor="black", hatch="o", label="3000<=value<5000")
)
df[df["value"] >= 5000].plot(
ax=ax,
color="r",
edgecolor="k",
hatch="*",
legend=False,
)
legend_list.append(mpatches.Patch(
facecolor="r",
edgecolor="black", hatch="*", label="value>=5000"))
ax.legend(handles = legend_list,
loc=(1, 0.5),
title="value 等级",
fontsize=12)
plt.show()
4. 总结
结合地理信息展示数据,让数据更加的生动,特别是很多数据分析场景本身就和地理信息密切相关。
除了基于地域的各种经济或者人口等等的相关数据,
还有各类气象数据,如能和地理信息结合展示的话,会让人对整体情况一目了然。
文中用到的主要资源:
- DataV数据可视化平台:https://datav.aliyun.com/portal/school/atlas/area_selector
- GeoPandas:https://github.com/geopandas/geopandas