我想根据索引列表对任意维d数据列的n轴重新排序顺序。如果轴是最后一个轴,那么这个问题(Reordering last dimension of a numpy ndarray)的解决方案就足够了。然而,在我的例子中,轴通常不是第一个或最后一个轴,因此省略本身并不能解决这个问题
这就是我目前提出的解决方案:
axes_list = list(range(d))
axes_list[0], axes_list[i] = axes_list[i], axes_list[0]
ndarr = np.transpose(ndarr, axes=axes_list)[order,...] # Switches the axis with the first and reorders
ndarr = np.transpose(ndarr, axes=axes_list) # Switches the axes back
我不喜欢这个解决方案的地方是我必须手动转置ndarray。我想知道是否有一个省略算子的推广,这样我们就可以解释一个选定数量的轴,这样
ndarr[GenEllipsis(n),order,...]
将跳过n第一个轴,并将(n+1)第个轴重新排序
这可能吗
使用命令np.take_along_axis并将输出分配给新变量。请参阅下面的代码:
@hpaulj的评论也是一个有效的答案(而且可能更好,请给他们一个投票权!)
相关问题 更多 >
编程相关推荐