python numpy array转置_NumPy数组的转置和轴对换

在 NumPy 中，利用 transpose()、ndarray.T、rollaxis() 和 swapaxes() 等函数可实现翻转数组的操作。

transpose() 函数

该函数可实现翻转给定数组的维度，并且返回数组的 1 个视图(View)。使用transpose()函数的方法如下：

numpy.transpose(arr)

其中，参数 arr 表示要转置的数组。例如 ar r是一个 3 行 5 列数组，numpy.transpose(arr) 函数就会返回 arr 数组转置为 5 行 3 列的数组视图，而原数组 arr 不改变。

ndarray.T 函数

该函数是属于 ndarray 类，作用类似于 numpy.transpose

rollaxis() 函数

该函数可实现向后滚动特定的轴，直到 1 个特定位置。使用 rollaxis() 函数的方法如下：

numpy.rollaxis(arr,axis,start)

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

swapaxes() 函数

该函数可交换数组的两个轴。NumPy1.10 之前的版本会返回交换后数组的视图。使用 swapaxes() 函数的方法如下：

numpy.swapaxes(arr,axis1,axis2)

其中，参数 arr 表示数组，axis1 表示对应第 1 个轴的整数，axis2 表示对应第 2 个轴的整数。

翻转数组的示例

示例代码 example1 如下。

# -*- coding: UTF-8 -*-

import numpy as np

#创建数组2行5列

arr = np.arange(10).reshape(2,5)

print('原数组：',arr)

#用transpose函数转置

print('转置数组：')

print(np.transpose(arr))

#用ndarray.T转置

print('转置数组：')

print(arr.T)

#用rollaxis函数

arr = np.arange(8).reshape(2,2,2)

print('原数组：',arr)

# 将轴 2 滚动到轴 0(宽度到深度)

print('调用rollaxis函数：')

print(np.rollaxis(arr,2))

# 将轴 0 滚动到轴 1(宽度到高度)

print('调用rollaxis函数：')

print(np.rollaxis(arr,2,1))

#用swapaxes函数

arr = np.arange(8).reshape(2, 2, 2)

print('原数组：',arr)

# 现在交换轴 0(深度方向)到轴 2(宽度方向)

print('调用swapaxes函数后的数组：')

print(np.swapaxes(arr, 2, 0))