Python numpy rollaxis用法及代碼示例

本文簡要介紹 python 語言中 numpy.rollaxis 的用法。

用法:

numpy.rollaxis(a, axis, start=0)

向後滾動指定的軸，直到它位於給定位置。

為了向後兼容，繼續支持此函數，但您應該更喜歡 moveaxis 。 moveaxis 函數是在 NumPy 1.11 中添加的。

參數：

a： ndarray

輸入數組。

axis： int

要滾動的軸。其他軸的位置相對於彼此不會改變。

start： 整數，可選

當 start <= axis 時，軸會回滾，直到它位於該位置。當 start > axis 時，軸會滾動，直到它位於該位置之前。默認值 0 會產生 “complete” 滾動。下表說明了如何解釋 start 的負值：

start	標準化start
-(arr.ndim+1)	提高AxisError
-arr.ndim	0
⋮	⋮
-1	arr.ndim-1
0	0
⋮	⋮
arr.ndim	arr.ndim
arr.ndim + 1	提高AxisError

返回：

res： ndarray

對於 NumPy >= 1.10.0 始終返回 a 的視圖。對於早期的 NumPy 版本，僅當軸的順序更改時才返回 a 的視圖，否則返回輸入數組。

例子：

>>> a = np.ones((3,4,5,6))
>>> np.rollaxis(a, 3, 1).shape
(3, 6, 4, 5)
>>> np.rollaxis(a, 2).shape
(5, 3, 4, 6)
>>> np.rollaxis(a, 1, 4).shape
(3, 5, 6, 4)