2D数组numpy列表的逐像素中值

Question

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

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

我想要：

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

我可以用for循环的“C风格”来实现这一点，最终cython或numba来加速这个过程，因为我的数组非常大，但我很确定我在numpy中缺少了一种简单而有效的方法？你知道吗

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

提前谢谢！你知道吗

Answer 1

可以使用^{}按深度顺序堆叠数组，并沿最后一个轴取^{}：

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

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