numpy.moveaxis()函数 | Python

numpy.moveaxis()函数可以将数组的轴移动到新的位置。

语法

numpy.moveaxis(arr, source, destination)

参数

• arr：要操作的输入数组。
• source：要移动的轴的原始位置，默认为0。
• destination：要移动轴的新位置，默认为0。

返回值

返回移动轴后的新数组。

示例

import numpy as np

# 定义一个 3x4x5 的三维数组
arr = np.random.rand(3, 4, 5)

# 移动轴，将原始位置为 0 的轴移动到新的位置为 2
new_arr = np.moveaxis(arr, 0, 2)

print('Original array shape:', arr.shape)
print('New array shape:', new_arr.shape)

输出结果：

Original array shape: (3, 4, 5)
New array shape: (5, 4, 3)

解释

在这个例子中，我们将原数组的第一个轴移动到了第三个轴的位置上。这就相当于将原数组从一个 3x4x5 的三维数组变换成了一个 5x4x3 的三维数组。