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)

總結

以上是生活随笔為你收集整理的Python squeeze()函数的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。