我真的有麻烦了。我有一个三维numpy数组,我想用一个二维索引数组对它重新排序。实际上,数组是通过编程确定的,三维数组可能是二维或四维的,但为了简单起见,如果两个数组都是二维的,下面是期望的结果:
ph = np.array([[1,2,3], [3,2,1]])
ph_idx = np.array([[0,1,2], [2,1,0]])
for sub_dim_n, sub_dim_ph_idx in enumerate(ph_idx):
ph[sub_dim_n] = ph[sub_dim_n][sub_dim_ph_idx]
这使得ph阵列变成:
array([[1, 2, 3],
[1, 2, 3]])
这正是我想要的。如果是同样的情况,有谁能帮忙,但我有一个三维数组(psh)而不是ph,比如:
psh = np.array(
[[[1,2,3]],
[[3,2,1]]]
)
希望这是清楚的,如果不是请询问。提前谢谢!你知道吗
所以,这些线索已经在这里有用的注释中了,但是为了完整性,它只需要使用np.沿\u轴取\u以及2d阵列的广播版本:
这样做的好处是,如果三维阵列的dim1包含多个元素,也可以工作:
这给了
如果您希望得到一个
ph.shape
形状的数组,您可以简单地np.squeeze
ph_ixs
使形状匹配,并使用它来索引ph
:相关问题 更多 >
编程相关推荐