温馨提示×

使用Matplotlib如何绘制3D图形

小亿
109
2024-05-20 12:49:33
栏目: 编程语言

要绘制3D图形,可以使用Matplotlib中的mplot3d模块。以下是一个简单的示例,展示如何绘制一个立方体:

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

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

# 定义立方体的顶点
vertices = [
    [0, 0, 0],
    [1, 0, 0],
    [1, 1, 0],
    [0, 1, 0],
    [0, 0, 1],
    [1, 0, 1],
    [1, 1, 1],
    [0, 1, 1]
]

# 定义立方体的面
faces = [
    [vertices[0], vertices[1], vertices[2], vertices[3]],
    [vertices[4], vertices[5], vertices[6], vertices[7]],
    [vertices[0], vertices[1], vertices[5], vertices[4]],
    [vertices[2], vertices[3], vertices[7], vertices[6]],
    [vertices[1], vertices[2], vertices[6], vertices[5]],
    [vertices[4], vertices[7], vertices[3], vertices[0]]
]

# 绘制立方体的面
for face in faces:
    x = [vertex[0] for vertex in face]
    y = [vertex[1] for vertex in face]
    z = [vertex[2] for vertex in face]
    ax.plot(x, y, z)

plt.show()

运行以上代码即可绘制一个简单的立方体。您也可以根据需要调整顶点和面的位置,以绘制不同的3D图形。

0