Python数据可视化-第7章-绘制3D图表和统计地图

环境

开发工具

VSCode

库的版本

numpy==1.26.4
matplotlib==3.10.1
ipympl==0.9.7

教材

本书为《Python数据可视化》一书的配套内容，本章为第7章 绘制3D图表和统计地图

本章首先介绍了使用mplot3d工具包绘制3D图表，然后介绍了使用animation模块制作动画，最后介绍了使用basemap工具包绘制统计地图。通过对本章的学习，希望读者能够掌握这些工具包和动画模块的基本用法。

参考

第7章-绘制3D图表和统计地图

7.1 使用mplot3d绘制3D图表

7.1.1 mplot3d概述

matplotlib不仅专注于二维图表的绘制，也具有绘制3D图表、统计地图的功能，并将这些功能分别封装到工具包mpl_toolkits.mplot3d、mpl_toolkits.basemap中，另外可以结合animation模块给图表添加动画效果。

mplot3d是matplotlib中专门绘制3D图表的工具包，它提供了一个重要的类Axes3D，该类继承自Axes类，使用Axes3D类可以构建一个三维坐标系的绘图区域。
通过以下两种方式可以创建Axes3D类的对象。

第一种：Axes3D()方法。

第二种：add_subplot()方法。

Axes3D()方法

Axes3D()是构造方法，它直接用于构建一个Axes3D类的对象。

Axes3D(fig, rect=None, *args, azim=-60, elev=30, zscale=None, 
sharez=None, proj_type='persp', **kwargs)

fig：表示绘图区域所属的画布。

rect：表示绘图区域的位置，它的值是一个元组(left, bottom, width, height)，元组中的前两个元素表示绘图区域左侧、底部到画布左侧、底部的距离与画布宽度、高度的比例，后两个元素表示绘图区域的宽度和高度与画布宽度、高度的比例。

azim ：表示方位角，默认值是-60度。

elev：表示仰角，默认值是30度。

proj_type：表示投影类型，该参数支持两种取值'persp'（透视图）和 'ortho'（正交视图），默认值为'persp'。

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
fig = plt.figure()
ax = Axes3D(fig)
plt.show()

add_subplot()方法

在调用add_subplot()方法添加绘图区域时为该方法传入projection='3d'，即指定坐标系的类型为三维坐标系，并返回一个Axes3D类的对象。

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
print(type(ax))
plt.show()

输出如下：

官方推荐使用add_subplot()方法创建Axes3D类的对象。

Axes3D类中提供了一些用于设置标题和坐标轴的方法，关于这些方法及说明具体如下表所示。

7.1.2 绘制常见的3D图表

常见的3D图表包括3D线框图、3D曲面图、3D柱形图、3D散点图等。 Axes3D类中提供了一些绘制3D图表的方法，关于这些方法及其说明如下表所示。

绘制3D线框图 plot_wireframe()方法

Axes3D类的对象使用plot_wireframe()方法绘制3D线框图。

plot_wireframe(self, X, Y, Z, *args, **kwargs)

X，Y，Z：表示x、y、z轴的数据。

rcount，ccount：表示每个轴方向上使用的最大样本量，默认为50。若输入的样本量更大，则会采用降采样的方式减少样本的数量；若输入的样本量为0，则不会对相应轴方向的数据进行采样。

rstride，cstride：表示采样的密度。若仅使用参数rstride或cstride中任意一个，则另一个参数默认为1。

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import axes3d
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
# 获取测试数据
X, Y, Z = axes3d.get_test_data(0.05)
# 绘制 3D线框图
ax.plot_wireframe(X, Y, Z, rstride=10, cstride=10)
plt.show()

输出如下:

绘制3D曲面图-plot_surface()方法

Axes3D类的对象使用plot_surface()方法绘制3D曲面图。

plot_surface(self, X, Y, Z, *args, norm=None, vmin=None, vmax=None, 
lightsource=None, **kwargs)

X，Y，Z：表示x、y、z轴的数据。

rcount，ccount：表示每个坐标轴方向上使用的最大样本量，默认为50。

rstride，cstride：表示采样的密度。

color：表示曲面的颜色。

cmap：表示曲面的颜色映射表。

shade：表示是否对曲面进行着色。注意，若使用cmap参数，则不可以使用shade参数。

linewidth ：表示线宽。

antialiased：表示是否抗锯齿。

from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
from matplotlib import cm
import numpy as np
x1 = np.arange(-5, 5, 0.25)
y1 = np.arange(-5, 5, 0.25)
print(x1.shape,"\n",x1)
print(y1.shape,"\n",y1)
print("---------")
x1, y1 = np.meshgrid(x1, y1)
print(x1.shape,"\n",x1)
print(y1.shape,"\n",y1)
print("---------")
r1 = np.sqrt(x1** 2 + y1 ** 2)
z1 = np.sin(r1)
print(r1.shape,"\n",r1)
print(z1.shape,"\n",z1)
print("---------")
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
# 绘制曲面图
ax.plot_surface(x1, y1, z1, cmap=cm.coolwarm, linewidth=0, antialiased=False) 
# 设置 z 轴刻度的范围、位置、格式
ax.set_zlim(-1.01, 1.01)
plt.show()

输出如下：

7.1.3 实例1：三维空间的星星

本实例要求根据一组测试数据，绘制包含若干个五角星的3D散点图。

# 01_stars_in_3d
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
plt.rcParams["font.sans-serif"] = ["SimHei"]
plt.rcParams["axes.unicode_minus"] = False
# 生成测试数据
x = np.random.randint(0, 40, 5)
y = np.random.randint(0, 40, 5)
z = np.random.randint(0, 40, 5)
# 创建三维坐标系的绘图区域, 并在该区域中绘制3D散点图
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
for xx, yy, zz in zip(x, y, z):
    print(xx, "  ",yy, "  ",zz)
    color = 'y'
    if 10 < zz < 20:
        color = '#C71585'
    elif zz >= 20:
        color = '#008B8B'
    ax.scatter(xx, yy, zz+3, c=color, marker='*', s=160, linewidth=1, edgecolor='black')
    ax.text(xx, yy, zz, f'({xx}, {yy}, {zz})', fontsize=10, ha='center', va='bottom')
ax.set_xlabel('x轴')
ax.set_ylabel('y轴')
ax.set_zlabel('z轴')
ax.set_title('3D散点图', fontproperties='simhei', fontsize=14)
plt.tight_layout()
plt.show()

输出如下：

38 11 22

0 9 10

20 3 29

15 31 16

29 2 23

7.2 使用animation制作动图

7.2.1 animation概述

matplotlib在1.1版本的标准库中加入了动画模块------animation，使用该模块的Animation类可以实现一些基本的动画效果。Animation类是一个动画基类。

FuncAnimation类

FuncAnimation是基于函数的动画类，它通过重复地调用同一函数来制作动画。

FuncAnimation(fig, func, frames=None, init_func=None, fargs=None, 
save_count=None, *, cache_frame_data=True, **kwargs)

fig：表示动画所在的画布。

func：表示每帧动画调用的函数，该函数会被matplotlib库自动调用，并在被调用时将frames中迭代的下一个值传递给它的第一个参数。

frames：表示动画的长度（一次动画包含的帧数），该参数的值是一个可迭代对象。

init_func：表示用于开始绘制帧的函数，它会在第一帧动画之前调用一次。若未设置该参数，则程序将使用frames 中第一项的绘图结果。

fargs：表示传递给func函数的其它参数。

interval：表示更新动画的频率，以毫秒为单位，默认为200。

blit：表示是否更新所有的点，默认为False。

# 以qt5为图形界面后端
# %matplotlib 可以使用的后端有：
# inline：将图形嵌入到Jupyter Notebook中
# notebook：将图形嵌入到Jupyter Notebook中
# qt5：在单独的窗口中显示图形
# wx：在单独的窗口中显示图形
# tk：在单独的窗口中显示图形
# osx：在单独的窗口中显示图形
# pdf：将图形保存为PDF文件
# svg：将图形保存为SVG文件
# ps：将图形保存为PS文件
# agg：将图形保存为AGG文件
# svgz：将图形保存为SVGZ文件

# %matplotlib widget
import numpy as np
import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation    # 导入动画类
x = np.arange(0, 2 *np.pi, 0.01)
fig, ax = plt.subplots()
line, = ax.plot(x, np.sin(x))
# 定义每帧动画调用的函数 
def animate(i):
    line.set_ydata(np.sin(x + i / 10.0))
    if(i%10==0):
        # ax.text(1+i%2, 0, i)
        print(i)
        pass
    return line
# 定义初始化帧的函数
def init():
    line.set_ydata(np.sin(x))
    return line
ani = FuncAnimation(fig=fig, func=animate, frames=100, 
                    init_func=init, interval=20, blit=False)
plt.show()

ArtistAnimation类

ArtistAnimation是基于一组Artist对象的动画类，它通过每帧绘制一个或一组Artist对象制作动画。

ArtistAnimation(fig, artists, interval, repeat_delay, repeat, 
blit, *args, **kwargs)

fig：表示动画所在的画布。

artists：表示一组Artist 对象的列表。

interval：表示更新动画的频率，以毫秒为单位，默认为200。

repeat_delay：表示再次播放动画之前延迟的时长。

repeat：表示是否重复播放动画。

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.animation import ArtistAnimation
x = np.arange(0, 2 * np.pi, 0.01)
fig, ax = plt.subplots()
arr = []
for i in range(5):
    line = ax.plot(x, np.sin(x + i))
    arr.append(line)

# 根据arr存储的一组图形创建动画
ani = ArtistAnimation(fig=fig, artists=arr, repeat=True)
plt.show()

大家希望将动画存储为视频文件，可以先安装ffmpeg或mencoder，之后使用Animation类的save()方法将每一帧动画存储为视频文件。

7.2.2 实例2：三维空间闪烁的星星

本实例要求在7.1.3绘制的3D散点图中添加动画，实现五角星由红色到白色最的闪烁效果 。

# 02_twinkling_stars_in_3d
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
from matplotlib.animation import FuncAnimation
plt.rcParams["font.sans-serif"] = ["SimHei"]
plt.rcParams["axes.unicode_minus"] = False
# 生成测试数据
xx = np.array([13, 5, 25, 13, 9, 19, 3, 39, 13, 27])
yy = np.array([4, 38, 16, 26, 7, 19, 28, 10, 17, 18])
zz = np.array([7, 19, 6, 12, 25, 19, 23, 25, 10, 15])
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
# 绘制初始的3D散点图
star = ax.scatter(xx, yy, zz, c='#C71585', marker='*', s=160, 
      linewidth=1, edgecolor='black')
# 每帧动画调用的函数
def animate(i):
    if i % 2:
        color = '#C71585'
    else:
        color = 'white'
    next_star = ax.scatter(xx, yy, zz, c=color, marker='*', s = 160, linewidth=1, edgecolor='black')
    return next_star
def init():
    return star
ani = FuncAnimation(fig=fig, func=animate, frames=None, init_func =init, interval=1000, blit=False)
ax.set_xlabel('x轴')
ax.set_ylabel('y轴')
ax.set_zlabel('z轴')
ax.set_title('3D散点图', fontproperties='simhei', fontsize=14)
plt.tight_layout()
plt.show()

7.3 使用basemap绘制统计地图

7.3.1 basemap概述

统计地图是以地图为底本，用各种几何图形、实物形象或不同线纹、颜色等表明指标的大小及其分布状况的图形，它是统计图形与地图的结合，既可以突出说明某些现象在地域上的分布，也可以对某些现象进行不同地区间的比较，还可以表明现象所处的地理位置及与其他自然条件的关系等。

basemap是matplotlib中用于绘制地图背景的工具包，它一般情况下不会参与绘图操作，而是将给定的地理坐标转换到地图投影上，之后将数据交给matplotlib进行绘图。

按照basemap

pip install basemap

到入basemap

from mpl_toolkits.basemap import Basemap

basemap工具包中提供了一个重要的类Basemap， 表示基础地图背景，创建Basemap类的对象时可以指定地图投影的类型和要处理的地球区域等一些内容。

Basemap(llcrnrlon=None, llcrnrlat=None, urcrnrlon=None, urcrnrlat=None, llcrnrx=None, ..., ax=None)

lon_0，lat_0：表示所需地图投影区域中心的经度或纬度。

llcrnrlon，llcrnrlat：表示地图投影区域左下角的经度或纬度。

urcrnrlon，urcrnrlat：表示地图投影区域右上角的经度或纬度。

width，height：表示所需地图投影区域的宽度和高度。

rsphere：表示投影中使用的球体的半径，默认值为6370997米。

resolution：表示包括海岸线、湖泊等的分辨率，可以取值为'c'（粗略，默认值）、'l'（低）、'i'（中级）、'h'（高）、'f'（完整）或None。

area_thresh：表示不会绘制海岸线或湖泊的阈值。

anchor：表示地图置于绘图区域的方式，默认为C，表示地图居中。

projection：表示地图投影的类型，默认值为cyl（等距圆柱投影）。

projection参数的常用取值及说明如下所示。

确定地图背景的投影区域之后，我们还需要对待处理的区域进行完善，比如在该区域中绘制河岸线、河流和地区或国家边界等。Basemap类中提供了一些绘制地图背景的方法。

拥有地图背景之后便可以使用matplotlib在地图上根据数据绘制图形。为方便用户操作，Basemap类中提供了一些在地图上绘制数据的方法（这些方法其实简单地转发到Axes实例方法，且进行了一些额外的处理和参数检查） 。

7.3.2 实例3：美国部分城镇人口分布

本实例要求根据下表的数据（存储于2014_us_cities.csv文件中），获取前500条数据，将获取的lat和lon两列的地理坐标转换到地图投影中，将pop列的数据绘制成气泡并显示到地图上。

# 03_twinkling_stars_in_3d
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
from mpl_toolkits.basemap import Basemap
import os
os.chdir(os.path.dirname(os.path.abspath(__file__)))
print(os.getcwd())
plt.rcParams["font.sans-serif"] = ["SimHei"]
plt.rcParams["axes.unicode_minus"] = False
# 创建 Basemap 对象
map = Basemap(projection='stere', lat_0=90, lon_0=-105, llcrnrlat=23.41, 
              urcrnrlat=45.44, llcrnrlon=-118.67, urcrnrlon=-64.52, 
              rsphere=6371200., resolution='l', area_thresh=10000)
map.drawmapboundary()     # 绘制地图投影周围边界
map.drawstates()          # 绘制州界
map.drawcoastlines()      # 绘制海岸线
map.drawcountries()       # 绘制国家边界
# 绘制纬线
parallels = np.arange(0., 90, 10.)
map.drawparallels(parallels, labels=[1, 0, 0, 0], fontsize=10)
# 绘制经线
meridians = np.arange(-110., -60., 10.)
map.drawmeridians(meridians, labels=[0, 0, 0, 1], fontsize=10)

posi = pd.read_csv("./素材/2014_us_cities.csv")
# 从3228组城市数据中选择500 组数据
lat = np.array(posi["lat"][0:500])               # 获取纬度值
lon = np.array(posi["lon"][0:500])               # 获取经度值
pop = np.array(posi["pop"][0:500], dtype=float)  # 获取人口数
# 气泡图的气泡大小
size = (pop / np.max(pop)) * 1000 
x, y = map(lon, lat)
map.scatter(x, y, s=size)
plt.title('2014年美国部分城镇的人口分布情况')
plt.show()

输出如下：