二维数组列表的逐像素中值 - numpy

Question

我有一个numpy 2D ndarray列表，我正在尝试计算列表中所有 dem 的“逐像素”中值和某个百分位数。 基本上这些是光栅，我需要保持它们的尺寸。 例如：

A = [[1,1],[2,2]]
B = [[2,2],[3,3]]
C = [[3,3],[4,4]]

而且我要：

my_median = [[2,2],[3,3]]

我可以在“C 风格”中使用 for 循环并最终使用cython或numba来加快进程，因为我的数组非常大，但我很确定我在numpy缺少一种简单有效的方法？

np.median(np.sum([A,B,C]))只给我全局中位数，或者如果我使用axis每行或每列的中位数。

提前致谢！

Answer 1

您可以使用dstack按顺序深度堆叠数组，并沿最后一个轴取np.median ：

np.median(np.dstack([A,B,C]), -1)

array([[2., 2.],
       [3., 3.]])