Skip to main content
 首页 » 编程设计

python之高效的二维numpy数组统计

2025年05月04日17bhlsheji

我有很多 100x100 的网格,是否有一种有效的方法使用 numpy 计算每个网格点的中值并只返回一个 100x100 的网格和中值?目前,我正在使用 for 循环遍历每个网格点,计算中位数,然后在最后将它们组合成一个网格。我确信有更好的方法可以使用 numpy 来做到这一点。任何帮助,将不胜感激!谢谢!

请您参考如下方法:

创建 100x100xN 数组(如果不可能,则堆叠在一起)并使用带有正确轴的 np.median 一次性完成:

import numpy as np 
a = np.random.rand(100,100) 
b = np.random.rand(100,100) 
c = np.random.rand(100,100) 
d = np.dstack((a,b,c)) 
result = np.median(d,axis=2)