使用matplotlib绘制圆锥体的三维图形

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

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

import numpy as np

theta = np.linspace(0, 2*np.pi, 100)
x = np.cos(theta)
y = np.sin(theta)

ax.plot(x, y, zs=0, zdir='z', label='Base')

k = 1  # 圆锥面的高度

# 绘制圆锥面
X = np.outer(np.linspace(-1, 1, 100), np.ones(100))
Y = X.copy()
Z = k * np.sqrt(X**2 + Y**2)
ax.plot_surface(X, Y, Z, cmap='viridis', alpha=0.5)

# 绘制顶点
ax.scatter([0], [0], [k], color='red', label='Apex')

ax.set_title('3D Cone')
ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('Z')
ax.legend()
plt.show()

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

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

# 绘制圆锥体底面
theta = np.linspace(0, 2*np.pi, 100)
x = np.cos(theta)
y = np.sin(theta)
ax.plot(x, y, zs=0, zdir='z', label='Base')

# 绘制圆锥体侧面
k = 1  # 圆锥面的高度
X = np.outer(np.linspace(-1, 1, 100), np.ones(100))
Y = X.copy()
Z = k * np.sqrt(X**2 + Y**2)
ax.plot_surface(X, Y, Z, cmap='viridis', alpha=0.5)
ax.scatter([0], [0], [k], color='red', label='Apex')

ax.set_title('3D Cone')
ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('Z')
ax.legend()
plt.show()