使用Python、Numpy和Matplotlib绘制掩蔽表面图

使用Python、Numpy和Matplotlib绘制掩蔽表面图，我们可以按照以下步骤进行操作 −

设置图形尺寸并调整子图之间和周围的填充。
创建一个新图形或激活现有图形。
将“ax”添加到图形作为子图布置的一部分。
从坐标向量pi和theta返回坐标矩阵。
使用掩蔽数据点创建x、y和z。
使用x、y和z数据点创建一个表面图。
要显示图形，请使用 show() 方法。

示例

import matplotlib.pyplot as plt
import numpy as np

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

fig = plt.figure()
ax = fig.add_subplot(111, projection="3d")
pi, theta = np.meshgrid(
   np.arange(1, 10, 2) * np.pi / 4,
   np.arange(1, 10, 2) * np.pi / 4)

x = np.cos(pi) * np.sin(theta)
y = np.sin(pi) * np.sin(theta)
z = np.ma.masked_where(x >= 0.01, y)

ax.plot_surface(x, y, z, color='red')

plt.show()