前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >Numpy--改变数组维度

Numpy--改变数组维度

作者头像
Max超
发布2019-12-03 15:55:08
7740
发布2019-12-03 15:55:08
举报

原文链接:https://blog.csdn.net/m0_37586991/article/details/79758168

参考文章链接:https://blog.csdn.net/m0_37586991/article/details/79758168

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

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

reshape()

ravel()

flatten()

用元组设置维度

transpose() 和 swapaxes()

resize()

下面将依次进行说明

代码语言:javascript
复制
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 函数(可用于一维变为多维,不能用于多维的维度间的交换,如无法将(3,1024,512)变为(1024,512,3))

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 个房间

代码语言:javascript
复制
2.ravel函数 可以将多维数组展平(也就是变回一维)

c = b.ravel()

print©

得到一维数组

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)

c = b.flatten()

print©

得到一维数组

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

4.用元组设置维度

代码语言:javascript
复制
直接用一个正整数元组来设置数组的维度

b.shape = (6,4)

print(b)

这种做法将直接改变所操作的数组,现在数组b变成了一个 6*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]

5.transpose函数将矩阵进行转置(可以用于多维度的维度交换,如:

np.transpose(a,1,2,0) 将a的维度由(3,1024,512)变为(1024,512,3))

同样可以用于多维度维度交换的还有:swapaxes(),如:

a.swapaxes(0,1)其实就是把矩阵中0,1两个轴对换一下维度由(3,1024,512)变为(1024,3,512)

d = b.transpose()

print(d)

得到 4*6 的多维数组

[ 0 4 8 12 16 20

1 5 9 13 17 21

2 6 10 14 18 22

3 7 11 15 19 23]

6.resize函数 和reshape函数的功能一样,但resize会直接修改所操作的数组

b.resize((2,12))

print(b)

得到 2*12 的两维数组

[ 0 1 2 3 4 5 6 7 8 9 10 11

12 13 14 15 16 17 18 19 20 21 22 23]

并且这一步不可以通过赋值来实现,如下所示:

e = b.resize((2,12))

print(e)

此时结果为:

None

本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2019-11-15 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档