编辑代码

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
 
# 创建一个新的3D图形
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
 
# 定义正方形的顶点
x = [0, 1, 1, 0, 0]
y = [0, 0, 1, 1, 0]
z = [0, 0, 0, 0, 0]  # z坐标保持为0，以形成2D正方形在Z=0平面上
 
# 绘制正方形边缘
ax.plot_surface(x, y, z, color='b')  # 使用plot_surface绘制，因为我们需要填充颜色
 
# 设置轴比例相等以保持正方形形状
ax.set_box_aspect([1, 1, None])
 
# 显示图形
plt.show()