如何对数组的一维进行操作？

Question

我来自 IDL，所以我最习惯使用带有显式指示的 for 循环。我读过 python 如何以不同的方式做事，你应该可以说

for thing in things:

我想不通的是，如果我有一个 4 维数组并且我想在数组的一维中执行操作，我该如何将结果保存到 4 维数组中并执行此操作'python' 方式。

我有一个时间、高度、纬度、经度的 4 维数组。我想使用 N=9 的运行均值 window 来平滑它。

这是我正在使用的代码：

KMCM_T = g.variables['temperature'][:,:,:,:]      #K
N = 9
T_bar_run = []

for idx, lon in enumerate(KMCM_lon):
    for idy, lat in enumerate(KMCM_lat):
        for idz, lev in enumerate(KMCM_levels):
            T_bar_run[:][idz][idy][idx] = np.convolve(KMCM_T[:,idz,idy,idx], np.ones((N,))/N, mode='same')

Answer 1

在这种特定情况下，您可能会使用 scipy.ndimage.convolve1d:

from scipy.ndimage import convolve1d

T_bar_run = convolve1d(KMCM_T, np.ones(N)/N, axis=0, mode='constant')

"numpy way of doing things" 正在避免循环，因为在数字应用程序中，解释循环的开销通常比其有效负载的成本相形见绌。这是通过依赖向量化函数来完成的，即将特定操作应用于其数组参数的每个单元格的函数。

许多这样的函数沿着一个或几个维度自然地起作用，这就是为什么您会经常遇到 axis 关键字参数。

如何对数组的一维进行操作？

How to operate on one dimension of an array?

python

arrays

numpy