3D-堆叠 2D 直方图

Question

我有一堆二维直方图（正方形2D numpy数组），我想像这样在3D中堆叠：

（图片来源：Cardenas，Alfredo E.等人。“膜介导的N-乙酰-L-色氨酸酰胺无辅助转运：动力学实验和模拟。” 《物理化学B》116.9（2012）：2739-2750。）

有没有人有好的想法怎样做这个？我已经尝试过 Python plot - stacked image slices 的等高线方法，但结果不是最佳的。

- user1873329

---

您能详细说明您当前设置存在的问题吗？同时，请分享您编写的代码和生成的图像。目前，您并没有提出具体的问题。 - David Zwicker

Answer 1

我建议您使用Axes3D.plot_surface绘制平面曲面，并使用facecolor参数进行着色，如下所示：

import numpy;
from matplotlib import pyplot;
from matplotlib import cm;
from mpl_toolkits.mplot3d import Axes3D;
pyplot.interactive(True);

# Creat mesh.
X = numpy.arange(-1, 1, 0.1);
Y = numpy.arange(-1, 1, 0.1);
X, Y = numpy.meshgrid(X, Y);

# Create some data to plot.
A = numpy.copy(X);
B = numpy.copy(Y);
C = numpy.sqrt(X**2 + Y**2);
D = numpy.cos(C);
# Normalize data for colormap use.
A -= numpy.min(A); A /= numpy.max(A);
B -= numpy.min(B); B /= numpy.max(B);
C -= numpy.min(C); C /= numpy.max(C);
D -= numpy.min(D); D /= numpy.max(D);

# Create flat surface.
Z = numpy.zeros_like(X);

# Plot
fig = pyplot.figure();
ax = fig.gca(projection='3d');
ax.plot_surface(X, Y, Z, rstride=1, cstride=1, facecolors = cm.coolwarm(A));
ax.plot_surface(X, Y, Z+0.1, rstride=1, cstride=1, facecolors = cm.coolwarm(B));
ax.plot_surface(X, Y, Z+0.2, rstride=1, cstride=1, facecolors = cm.coolwarm(C));
ax.plot_surface(X, Y, Z+0.3, rstride=1, cstride=1, facecolors = cm.coolwarm(D));

- Bence

Answer 2

补充@Bence的答案，可以使用here中的代码片段添加额外的垂直色条。 具体来说，我使用了以下代码，其中“xxx”是我的数据集，在Bence的图中显示的每个层都是dstack'd numpy数组中的单独z层。

m = cm.ScalarMappable(cmap=cm.jet)
m.set_array(xxx)
pyplot.colorbar(m)
pyplot.show()

- AaronJPung