Matplotlib绘制三维图形

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(-1, 10, 20)
y = np.linspace(-1, 10, 20)
x, y = np.meshgrid(x, y)
z = x * y

fig = plt.figure(figsize=(9, 3))
ax1 = fig.add_subplot(121, projection='3d')
ax2 = fig.add_subplot(122, projection='3d')
surf1 = ax1.plot_surface(x, y, z, cmap = 'plasma')   # cmap指定颜色映射plasma
ax2.plot_surface(x, y, z)

# 设置坐标轴刻度
ax1.set_xticks([0, 5, 10])
ax1.set_xticks(range(0, 11, 1), minor=True)
ax1.set_yticks([0, 5, 10])
ax1.set_yticks(range(0, 11, 1), minor=True)
ax1.set_zticks([0, 100])
ax1.set_zticks(range(0, 101, 10), minor=True)
# ax1.set_xticklabels([])
# ax1.set_yticklabels([])
ax1.set_zticklabels([])    # 隐藏z轴刻度标签
# 添加色条
fig.colorbar(surf1, shrink=0.7, aspect=15)           # shrink控制色条的长度，aspect控制色条的宽度

plt.show()