Python squeeze()函数

最新推荐文章于 2025-11-04 11:00:00 发布

原创最新推荐文章于 2025-11-04 11:00:00 发布 · 4.2w 阅读

39 ·

CC 4.0 BY-SA版权

文章标签：

#python #squeeze

python 专栏收录该内容

15 篇文章

订阅专栏

本文详细介绍了numpy库中的squeeze()函数，该函数用于移除数组中所有长度为1的维度。文章通过实例展示了如何使用该函数，并解释了在指定axis参数时可能遇到的错误情况。

部署运行你感兴趣的模型镜像

Python squeeze()函数

作用:移除数组中维度为1的维度

函数形式:numpy.``squeeze(a, axis=None)

参数:axis: 选择数组中的某一维度移除, 如果选择形状输入大于1的轴，则会引发错误。

x = np.array([[[0], [1], [2]]])
x.shape
# (1, 3, 1)
np.squeeze(x).shape
# (3,)
np.squeeze(x, axis=0).shape
# (3, 1)
np.squeeze(x, axis=1).shape
'''
---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-31-182a4f2e2f87> in <module>
----> 1 np.squeeze(x, axis=1).shape

~/anaconda3/lib/python3.6/site-packages/numpy/core/fromnumeric.py in squeeze(a, axis)
   1290         return squeeze()
   1291     else:
-> 1292         return squeeze(axis=axis)
   1293 
   1294 def diagonal(a, offset=0, axis1=0, axis2=1):

ValueError: cannot select an axis to squeeze out which has size not equal to one
'''
np.squeeze(x, axis=2).shape
#(1, 3)

您可能感兴趣的与本文相关的镜像