标签:python 维度 shape np squeeze numpy axis
numpy.squeeze() 这个函数的作用是去掉矩阵里维度为1的维度。例如,(1, 5)的矩阵经由np.squeeze处理后变成5;(5, 1, 6)的矩阵经由np.squeeze处理后变成(5, 6)。
numpy提供了numpy.squeeze(a, axis=None)函数,从数组的形状中删除单维条目。其中a表示输入的数组;axis用于指定需要删除的维度,但是指定的维度必须为单维度,否则将会报错。axis的取值可为None 或 int 或 tuple of ints,若axis为空时则删除所有单维度的条目。
examples1:
>>> import numpy as np
>>> a = np.array([[[0], [1], [2]]])
>>> a.shape
(1, 3, 1)
>>> np.squeeze(a).shape
(3,)
>>> np.squeeze(a, axis=(2,)).shape
(1, 3)
>>> np.squeeze(a, axis=(0,)).shape
(3, 1)examples2:
>>> import numpy as np
>>> a = np.arange(5).reshape(1,5)
>>> a
array([[0, 1, 2, 3, 4]])
>>> a.shape
(1, 5)
>>> b = np.squeeze(a)
>>> b
array([0, 1, 2, 3, 4])
>>> b.shape
(5, )examples3:
>>> import numpy as np
>>> a = np.arange(10).reshape(1,2,5)
>>> a
array([[[0, 1, 2, 3, 4],
[5, 6, 7, 8, 9]]])
>>> a.shape
(1, 2, 5)
>>> b = np.squeeze(a)
>>> b
array([[0, 1, 2, 3, 4],
[5, 6, 7, 8, 9]])
>>> b.shape
(2, 5)
标签:python,维度,shape,np,squeeze,numpy,axis 来源: https://blog.51cto.com/u_15178976/2729138
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。