學習筆記193—Python squeeze()函數

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)

參考鏈接：https://blog.csdn.net/happyday_d/article/details/86016661