如何使用Matplotlib从灰度图创建曲面图？

要使用matplotlib从灰度图创建曲面图，我们可以按照以下步骤进行−

• 设置图片大小并调整子图之间和周围的填充。

• 使用Numpy创建随机数据点。

• 从2d图像数据栅格中获取 xx 和 yy 数据点。

• 创建一个新的图形或激活一个现有的图形。

• 获取绘图的当前轴并将其作为三维投影轴。

• 使用 cmap =’gray’ 创建一个曲面图。

• 要显示图形，请使用 show() 方法。

示例

import numpy as np
import matplotlib.pyplot as plt

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

data = np.random.rand(5, 5)

xx, yy = np.mgrid[0:data.shape[0], 0:data.shape[1]]

fig = plt.figure()

ax = fig.gca(projection='3d')
ax.plot_surface(xx, yy, data, rstride=1, cstride=1, linewidth=0, cmap='gray')

plt.show()

输出