python在for循环中绘制多个3D子图

有时候我们需要绘制好几个图代码，手动写每个subplot的代码，然而这样代码重复量太大。而绘图部分代码大部分相同甚至完全一样，能不能用for循环的方式呢？

开始进入正题，代码如下：

import numpy as np
import matplotlib.pyplot as plt
import matplotlib as mpl
from mpl_toolkits.mplot3d import Axes3D

# 设置属性防止中文乱码
mpl.rcParams['font.sans-serif'] = [u'SimHei']
mpl.rcParams['axes.unicode_minus'] = False

# 生成数据，为了简单起见，后续3个图的数据完全一样
radius = 1
Vz = 1
t = np.linspace(0, 20, 1000, endpoint=False)
x = radius * np.cos(2 * np.pi * t)
y = radius * np.sin(2 * np.pi * t)
z = Vz * t

# 画图
fig = plt.figure(figsize=(12, 6), facecolor='w')
for i in range(3):
    # 绘制多个子图
    ax = fig.add_subplot(1, 3, i + 1, projection='3d')
    if i == 0:		###################其实这3个图是一样的，为了加以区分，人为加入了判断条件，实际操作中无须如此
        ax.scatter3D(x, y, z, color='blue', marker='o', label='图例')  # 绘制散点图
        ax.plot3D(x, y, z, 'gray')  # 绘制空间曲线图
    elif i == 1:
        ax.scatter3D(x, y, z, color='red', marker='s', label='图例')
        ax.plot3D(x, y, z, 'gray')  # 绘制空间曲线图
    elif i == 2:
        ax.scatter3D(x, y, z, color='black', marker='x', label='图例')
        ax.plot3D(x, y, z, 'gray')  # 绘制空间曲线图
    plt.legend()

    # 坐标轴标签，标题
    ax.set_xlabel('X')
    ax.set_ylabel('Y')
    ax.set_zlabel('Z')
    fig.suptitle('结果示意图')  # 给图片添加标题
    ax.set_title(f'第{i + 1}个子图')
plt.show()

效果如下：