Python在numpy中生成3D球体

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

x = r * sin(theta) * cos(phi)
y = r * sin(theta) * sin(phi)
z = r * cos(theta)

radius = 1  # 球体半径
theta_steps = 50  # 极角步长
phi_steps = 100  # 方位角步长

theta = np.linspace(0, np.pi, theta_steps)
phi = np.linspace(0, 2 * np.pi, phi_steps)

theta, phi = np.meshgrid(theta, phi)

x = radius * np.sin(theta) * np.cos(phi)
y = radius * np.sin(theta) * np.sin(phi)
z = radius * np.cos(theta)

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.plot_surface(x, y, z, color='b')

ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('Z')

plt.show()