沿轴重新排序的numpy函数

2024-04-19 14:45:45 发布

您现在位置：Python中文网/ 问答频道 /正文
9066 3

有没有更简单的方法（np函数）来实现以下目标bb是我正在寻找的输出

import numpy as np
aa = np.arange(4*4*3).reshape(4,4,3)
bb = np.stack((aa[:,:,2],aa[:,:,1],aa[:,:,0]),axis=2)

我认为np.roll在这里不适用，因为它似乎是按顺序移动的


1条回答
网友
1楼 · 发布于 2024-04-19 14:45:45

在索引时，可以使用-1作为最后一个轴上的步长参数，沿该轴创建反向数组：

In [10]: bb = aa[:,:,::-1]

相关问题 更多 >

    编程相关推荐