numpy的rollaxis与swapaxes函数

numpy.rollaxis 函数向后滚动特定的轴到一个特定位置，格式如下：

numpy.rollaxis(arr, axis, start)

参数说明：

• arr：数组
• axis：要向后滚动的轴，其它轴的相对位置不会改变
• start：默认为零，表示完整的滚动。会滚动到特定位置。

import numpy as np

# 创建了三维的 ndarray
a = np.arange(27).reshape((3, 3, 3))

print('原数组：')
print(a)
print('\n')

print('输出结果：')
# 将轴 2 滚动到轴 1
print(np.rollaxis(a, 2, 1))

np.rollaxis(a, 2, 1) 轴2 插入 到 轴1 前面。

numpy.swapaxes 函数用于交换数组的两个轴，格式如下：

numpy.swapaxes(arr, axis1, axis2)

• arr：输入的数组
• axis1：对应第一个轴的整数
• axis2：对应第二个轴的整数

import numpy as np

# 创建了三维的 ndarray
a = np.arange(27).reshape((3, 3, 3))

print('原数组：')
print(a)
print('\n')

# 交换轴 2 到轴 0
print('输出结果：')
print(np.swapaxes(a, 2, 0))

np.swapaxes(a, 2, 0)  轴2 和 轴0 进行交互