如何通过 Python 获取 3D 彩色表面?

matplotlibpythondata visualization更新于 2024/3/1 16:12:17

要通过 Python 获取 3D 彩色表面，我们可以采取以下步骤 −

步骤

设置图形大小并调整子图之间和周围的填充。
使用 numpy 创建 x 和 y 数据点。
获取 3D 数据，即 z。
创建新图形或激活现有图形。
获取 3D 轴。
创建表面图。
要显示图形，请使用 show() 方法。

示例

import numpy as np
import matplotlib.pyplot as plt

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

x = np.linspace(-3, 3, 100)
y = np.cos(x)

x, y = np.meshgrid(x, y)

z = x ** 2 + y ** 2 - 2

fig = plt.figure()

ax = plt.axes(projection='3d')

surf = ax.plot_surface(x, y, z,
cmap=plt.get_cmap('hot'),
edgecolor='none')

plt.show()