python - 将颜色图添加到 add_collection(3d 图中的表面)

Question

我正在制作一个 3 边金字塔我使用 add_collection3d(Poly3DCollection

但我想添加一个基于 z 轴的颜色图，以便顶部为红色并逐渐变为蓝色，但细节无关紧要。我尝试了 cmap 选项，但它确实改变了颜色。

我在一侧添加了一个脚本来说明问题。

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

fig = plt.figure()
ax = Axes3D(fig)

#Pyramid side 1
xp1 = [1.1184,2.46,2.46]
yp1 = [0.5592, 1.23,2.73]
zp1 = [-1.5,0,-1.5]
verts1= [zip(xp1,yp1,zp1)]

cmhot = plt.cm.get_cmap("hot")
ax.add_collection3d(Poly3DCollection(verts1,cmap=cmhot))

ax.set_xlim(0,5)
ax.set_ylim(0,5)
ax.set_zlim(-2.5,5)
plt.show()

为什么这不起作用？

非常感谢。

Answer 1

你必须将 zip 转换成列表格式

list(zip(xp1,yp1,zp1))