numpy数组中reshape和squeeze函数的使用

本文详细介绍了numpy中的reshape和squeeze函数。reshape用于改变数组的轴和维度,而squeeze则用于删除形状中的单维度条目。squeeze函数的axis参数可以指定删除的单维度,不修改原数组。在matplotlib画图时,非单维数组可能引发问题,squeeze函数可以帮助调整合适维度。
摘要由CSDN通过智能技术生成

参考了:http://blog.csdn.net/zenghaitao0128/article/details/78512715,作了一些自己的补充。

numpy中的reshape函数和squeeze函数是深度学习代码编写中经常使用的函数,需要深入的理解。

其中,reshape函数用于调整数组的轴和维度,而squeeze函数的用法如下,

语法:numpy.squeeze(a,axis = None)

 1)a表示输入的数组;
 2)axis用于指定需要删除的维度,但是指定的维度必须为单维度,否则将会报错;
 3)axis的取值可为None 或 int 或 tuple of ints, 可选。若axis为空,则删除所有单维度的条目;
 4)返回值:数组
 5) 不会修改原数组;
作用:从数组的形状中删除单维度条目,即把shape中为1的维度去掉
 

举例:

numpy的reshape和squeeze函数:

import numpy as np
e = np.arange(10)
print(e)
一维数组:[0 1 2 3 4 5 6 7 8 9]
f = e.reshape(1,1,10)
print(f)

三维数组:(第三个方括号里有十个元素)

[[[0 1 2 3 4 5 6 7 8 9]]],前两

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值