python - 带有扭曲颜色图的 3D 图

Question

一个例子(Matlab)代码如下:

Z = peaks(80); 
figure;     surf(Z);        colormap(jet)
figure;     imshow(mod(Z,pi),[],'i','f');        colormap(jet)

它产生:

• 数据 Z 的 3D 图
• 2D View ，我将 Z 变形为 $\pi$:

我想将两者结合起来，即 warped color map:在 3D 绘图中可视化 Z，但使用 warped colormap 为其着色(即它的高度应该引用 3D 图，但它的颜色映射到 2D 图)。

如何在 Matlab 或 Python 中完成？

Answer 1

只需将颜色应基于的值作为第四个输入传递给 surf :

Z = peaks(80);
surf(1:size(Z,2), 1:size(Z,1), Z, mod(Z,pi))
colormap(jet)