Numpy原位转置

时间:2018-08-30 14:40:01

标签: python numpy

我有一个大的numpy ndarray形(H,W,3)

我需要这样做

new_arr = arr[:, :, [2,1,0]]

我计时了,这花了几毫秒。我的理解是,这是副本。是否有更快的放置操作?

1 个答案:

答案 0 :(得分:0)

我最近制作了一个库,用于执行2d,3d和4d数组的原位转置。对于正方形矩阵,就地转置比numpy快。否则,它的内存效率更高,但速度较慢。

https://github.com/seung-lab/fastremap