编程语言
首页 > 编程语言> > python-使用线性插值调整numpy ndarray的大小

python-使用线性插值调整numpy ndarray的大小

作者:互联网

假设我想使用线性插值将形状数组(100,100,100)调整为形状数组(57,57,57).

基本上,我需要一个函数,该函数采用形状为S的n维数组,并通过内插将其变换为不带有维数但形状为S’的数组.

是否有一种简单快速的方法来用numpy和scipy做到这一点?
我发现了诸如1d插值,2d插值,网格插值之类的东西,但是它们需要线性空间,而我并不是很了解.

解决方法:

您可以使用scipy.interpolate中的griddata方法

from scipy.interpolate import griddata

可以说您的3D阵列由以下公式给出:

array3d

您将需要重塑数据并使用原始索引或点坐标创建另一个数组

N=100
array3DAux = array3D.reshape(N**3)

# ijk is an (N**3,3) array with the indexes of the reshaped array.
ijk = mgrid[0:N,0:N,0:N].reshape(3,N**3).T

现在,您可以创建要查找新的内插点的新网格.

#In your case 57 points
n = 57j
i,j,k = mgrid[0:N:n,0:N:n,0:N:n]

有3种最接近,线性和三次插值方法
如下

newArray3D_z0 = griddata(ijk,array3DAux,(i,j,k),method="nearest")
newArray3D_z1 = griddata(ijk,array3DAux,(i,j,k),method="linear")
newArray3D_z2 = griddata(ijk,array3DAux,(i,j,k),method="cubic")

在这种情况下,可以减小或增加3D阵列的大小.

希望能帮助到你

标签:scipy,arrays,python,numpy,interpolation
来源: https://codeday.me/bug/20191122/2056071.html