使用Python创建N维网格

时间:2017-10-28 16:05:07

标签: python arrays numpy mesh

我试图为需要和理解距离的算法创建一个坐标网格。我知道如何针对已知数量的维度执行此操作 - 例如对于2D:

x = [0,1,2]
y = [10,11,12]


z = np.zeros((3,3,2))
for i,X in enumerate(x):
    for j,Y in enumerate(y):
        z[i][j][0] = X
        z[i][j][1] = Y
print(z)
--------------------------
array([[[  0.,  10.],
        [  0.,  11.],
        [  0.,  12.]],

       [[  1.,  10.],
        [  1.,  11.],
        [  1.,  12.]],

       [[  2.,  10.],
        [  2.,  11.],
        [  2.,  12.]]])

这很好用。我最终得到一个(3,3,2)的形状,其中2是该点坐标的值。我试图用它来创建一个概率曲面,所以我需要能够让每一个点成为它自己的位置"值。有没有办法轻松将其扩展为N维?在那里,我将有一个未知数量的for循环。由于项目限制,我可以访问Python内置函数和numpy,但这或多或少都有。

我尝试了np.meshgrid(),但它导致输出形状为(2,3,3),我尝试重塑它从未以正确的顺序给我坐标。关于如何干净利落地做任何想法?

1 个答案:

答案 0 :(得分:1)

我可以使用

复制您的z
In [223]: np.stack([np.tile([x],(1,3)).reshape(3,3).T,np.tile([y],(3,1))],2)
Out[223]: 
array([[[ 0, 10],
        [ 0, 11],
        [ 0, 12]],

       [[ 1, 10],
        [ 1, 11],
        [ 1, 12]],

       [[ 2, 10],
        [ 2, 11],
        [ 2, 12]]])

瓷砖片看起来像

In [224]: np.tile([y],(3,1))
Out[224]: 
array([[10, 11, 12],
       [10, 11, 12],
       [10, 11, 12]])
In [225]: np.tile([x],(1,3)).reshape(3,3).T
Out[225]: 
array([[0, 0, 0],
       [1, 1, 1],
       [2, 2, 2]])

我可以清理第二个。但基本思想是以stack可以将它们组合到所需的(n,n,2)数组中的方式复制输入。

一旦理解了这一点,就不难将事物扩展到3d及以上。但我还没有完全处理你的意图。

可能更简单(repeattile更快):

np.stack([np.repeat(x,3).reshape(3,3), np.repeat(y,3).reshape(3,3).T], 2)

对于更多维度,transpose可能需要细化。

meshgrid相同(可能在内部使用repeattile

In [232]: np.stack(np.meshgrid(x,y, indexing='ij'),2)
Out[232]: 
array([[[ 0, 10],
        [ 0, 11],
        [ 0, 12]],

       [[ 1, 10],
        [ 1, 11],
        [ 1, 12]],

       [[ 2, 10],
        [ 2, 11],
        [ 2, 12]]])

更高维度:

In [237]: np.stack(np.meshgrid([1,2], [10,20,30], [100,200,300,400], indexing='ij'), 3).sum(axis=-1)
Out[237]: 
array([[[111, 211, 311, 411],
        [121, 221, 321, 421],
        [131, 231, 331, 431]],

       [[112, 212, 312, 412],
        [122, 222, 322, 422],
        [132, 232, 332, 432]]])