我试图为需要和理解距离的算法创建一个坐标网格。我知道如何针对已知数量的维度执行此操作 - 例如对于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),我尝试重塑它从未以正确的顺序给我坐标。关于如何干净利落地做任何想法?
答案 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及以上。但我还没有完全处理你的意图。
可能更简单(repeat
比tile
更快):
np.stack([np.repeat(x,3).reshape(3,3), np.repeat(y,3).reshape(3,3).T], 2)
对于更多维度,transpose
可能需要细化。
与meshgrid
相同(可能在内部使用repeat
或tile
:
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]]])