NumPy 教程 之 Numpy 数组操作 8

Numpy 数组操作

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

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

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

numpy.rollaxis

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

numpy.rollaxis(arr, axis, start)

参数说明：

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

实例

import numpy as np

创建了三维的 ndarray

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

print ('原数组：')
print (a)
print ('获取数组中一个值：')
print(np.where(a==6))
print(a[1,1,0]) # 为 6
print ('\n')

将轴 2 滚动到轴 0（宽度到深度）

print ('调用 rollaxis 函数：')
b = np.rollaxis(a,2,0)
print (b)

查看元素 a[1,1,0]，即 6 的坐标，变成 [0, 1, 1]

最后一个 0 移动到最前面

print(np.where(b==6))
print ('\n')

将轴 2 滚动到轴 1：（宽度到高度）

print ('调用 rollaxis 函数：')
c = np.rollaxis(a,2,1)
print (c)

查看元素 a[1,1,0]，即 6 的坐标，变成 [1, 0, 1]

最后的 0 和 它前面的 1 对换位置

print(np.where(c==6))
print ('\n')

输出结果如下：

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

[[4 5]
[6 7]]]
获取数组中一个值：
(array([1]), array([1]), array([0]))
6

调用 rollaxis 函数：
[[[0 2]
[4 6]]

[[1 3]
[5 7]]]
(array([0]), array([1]), array([1]))

调用 rollaxis 函数：
[[[0 2]
[1 3]]

[[4 6]
[5 7]]]
(array([1]), array([0]), array([1]))