python - 在特定轴上调整 ndarray 的大小和插值

Question

假设我有这样的 3-d numpy 数组:

arr = np.random.randn(14,10,10)

我需要将其大小调整为 (14, 20, 20) 的形状。这意味着第一个维度是从 (10, 10) 单独调整大小的。至(20, 20)采用插值法。

我怎样才能做到这一点？

Answer 1

对于此类任务，一个方便的工具是 scipy.ndimage.interpolation.zoom

这将调整数组大小并使用 spline interpolation 对其进行插值。为了使用它，您需要提供一个缩放因子，在本例中应该是[1,2,2]。这将指定您希望沿最后两个轴的缩放系数为 2:

from scipy.ndimage import interpolation

arr = np.random.randn(14,10,10)
new_arr = interpolation.zoom(arr,[1,2,2])

print(new_arr.shape)
# (14, 20, 20)