带有扭曲色图的3D绘图

Question

示例（Matlab）代码如下：

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

它生成：

• 数据的Z三维图
• 2D视图，我将Z转换为$ \ pi $：

我想将两者结合起来，即扭曲的颜色贴图：在3D绘图中可视化Z，但使用扭曲的颜色贴图将其着色（即其高度应参考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)