python改变数组维度_Numpy 改变数组维度的几种方法小结

来自 《Python数据分析基础教程：Numpy 学习指南(第2版)》

Numpy改变数组维度的方法有：

reshape()

ravel()

flatten()

用元组设置维度

transpose()

resize()

下面将依次进行说明

0. 首先，创建一个多维数组

from numpy import *

a = arange(24)

得到：

[ 0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23]

1.reshape 函数

b = a.reshape(2,3,4)

print(b)

得到一个 2*3*4 维的数组：

[[[ 0 1 2 3]

[ 4 5 6 7]

[ 8 9 10 11]]

[[12 13 14 15]

[16 17 18 19]

[20 21 22 23]]]

可以想象成2层小洋楼，每层有 3*4 个房间

2.ravel函数 可以将多维数组展平(也就是变回一维)

c = b.ravel()

print(c)

得到一维数组

[ 0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23]

3.flatten函数 也是将多维数组展平，与ravel函数的功能相同，不过flatten函数会请求分配内存来保存结果，而ravel函数只是返回数组的一个视图(view)