matplotlib绘制3D金字塔图像

Matplotlib是一个广泛使用的Python数据可视化库，它支持创建各种类型的图表，包括三维图形。要在Matplotlib中绘制三维金字塔图像，我们可以使用`mplot3d`模块下的功能。以下是一个基本的例子：

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d.art3d import Poly3DCollection

# 定义金字塔的底面和侧面顶点坐标
vertices = np.array([
    [0, 0, 0],   # 底部左上角
    [1, 0, 0],   # 底部右上角
    [1, 1, 0],   # 底部右下角
    [0, 1, 0],   # 底部左下角
    [0, 0, 1],   # 顶部中心
])

faces = [
    [[0, 1, 2, 3]],   # 底部四边形
    [[4, 0, 2, 3]]    # 侧面三角形
]

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

# 创建Pyramid对象并设置颜色
pyramid = Poly3DCollection(faces, facecolors='tab:orange', edgecolor='k')
ax.add_collection3d(pyramid)

ax.set_xlim(-0.5, 1.5)
ax.set_ylim(-0.5, 1.5)
ax.set_zlim(0, 1)

plt.show()