python - 使用更有效的方法处理数据，例如 python 中的并行

Question

我有一组数据 U，其维度为 [ns, ny, nx]，单位为 sigma 级，并希望将其插入另一个标准压力级 ([nz,ny,nx])，这样我必须逐个网格处理这个问题，如下所示：

U2 = np.empty((nz, ny, nx))
for ix in range(NX):
  for iy in range(NY):
    tmp = np.interp(plev_new, plev_old[:,iy,ix] U[:,iy,ix], left=filled_value, right=filled_value)
    U2[:,iy,ix] = tmp

但是，我发现它在逐个网格处理数据时效率很低。我认为更快的方法是在 x 和 y 网格中并行。但是我在模块多处理中总是没能做到这一点。

对多处理或其他更好或更有效的方式的任何建议。

非常感谢

标签： pythonparallel-processing