python - 使用 Matplotlib 绘制椭圆体

Question

有人有绘制椭球体的示例代码吗？ matplotlib 站点上有一个球体，但没有椭圆体。我正在尝试绘制

x**2 + 2*y**2 + 2*z**2 = c

其中 c 是定义椭圆体的常量(如 10)。我尝试了 meshgrid(x,y) 路线，修改了方程式，因此 z 在一侧，但 sqrt 是个问题。 matplotlib sphere example works with angles, u,v，但我不确定如何为椭圆体工作。

Answer 1

这是通过球坐标实现的方法:

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

fig = plt.figure(figsize=plt.figaspect(1))  # Square figure
ax = fig.add_subplot(111, projection='3d')

coefs = (1, 2, 2)  # Coefficients in a0/c x**2 + a1/c y**2 + a2/c z**2 = 1 
# Radii corresponding to the coefficients:
rx, ry, rz = 1/np.sqrt(coefs)

# Set of all spherical angles:
u = np.linspace(0, 2 * np.pi, 100)
v = np.linspace(0, np.pi, 100)

# Cartesian coordinates that correspond to the spherical angles:
# (this is the equation of an ellipsoid):
x = rx * np.outer(np.cos(u), np.sin(v))
y = ry * np.outer(np.sin(u), np.sin(v))
z = rz * np.outer(np.ones_like(u), np.cos(v))

# Plot:
ax.plot_surface(x, y, z,  rstride=4, cstride=4, color='b')

# Adjustment of the axes, so that they all have the same span:
max_radius = max(rx, ry, rz)
for axis in 'xyz':
    getattr(ax, 'set_{}lim'.format(axis))((-max_radius, max_radius))

plt.show()

结果图类似于

上面的程序实际上生成了一个更好看的“方形”图形。

此解决方案的灵感来自 example在 Matplotlib's gallery .