NumPy 教程 之 Numpy 数组操作 9

Numpy 数组操作

Numpy 中包含了一些函数用于处理数组，大概可分为以下几类：

修改数组形状
翻转数组
修改数组维度
连接数组
分割数组
数组元素的添加与删除

修改数组形状
函数 描述
reshape 不改变数据的条件下修改形状
flat 数组元素迭代器
flatten 返回一份数组拷贝，对拷贝所做的修改不会影响原始数组
ravel 返回展开数组

numpy.swapaxes

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

numpy.swapaxes(arr, axis1, axis2)

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

实例

import numpy as np

创建了三维的 ndarray

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

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

现在交换轴 0（深度方向）到轴 2（宽度方向）

print ('调用 swapaxes 函数后的数组：')
print (np.swapaxes(a, 2, 0))

输出结果如下：

原数组：
[[[0 1]
[2 3]]

[[4 5]
[6 7]]]

调用 swapaxes 函数后的数组：
[[[0 4]
[2 6]]

[[1 5]
[3 7]]]