Лучший способ интерполировать numpy.ndarray вдоль оси ⇐ Python

[Anonymous]

У меня есть четырехмерные данные, скажем, о температуре, в numpy.ndarray.
Форма массива: (ntime, nheight_in, nlat, nlon).

У меня есть соответствующие одномерные массивы для каждого из измерений, которые скажите мне, какому времени, высоте, широте и долготе соответствует определенное значение, для этого примера мне нужен height_in, задающий высоту в метрах.

Теперь я нужно перенести его на другое измерение высоты, height_out с другой длиной.

Похоже, что следующее делает то, что я хочу:

Код: Выделить всё

ntime, nheight_in, nlat, nlon = t_in.shape

nheight_out = len(height_out)
t_out = np.empty((ntime, nheight_out, nlat, nlon))

for time in range(ntime):
for lat in range(nlat):
for lon in range(nlon):
t_out[time, :, lat, lon] = np.interp(
height_out, height_in, t[time, :, lat, lon]
)

Но с тремя вложенными циклами и множеством переключений между Python и numpy я не думаю, что это лучший способ сделать это.

Есть предложения, как это улучшить? Спасибо

Подробнее здесь: https://stackoverflow.com/questions/289 ... ng-an-axis