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()
Matplotlib绘制三维图形
最新推荐文章于 2024-05-16 01:12:30 发布