當前位置: 首頁>>編程示例 >>用法及示例精選 >>正文


Python dask.array.squeeze用法及代碼示例

用法:

dask.array.squeeze(a, axis=None)

a 中刪除長度為 1 的軸。

此文檔字符串是從 numpy.squeeze 複製的。

可能存在與 Dask 版本的一些不一致之處。

參數

aarray_like

輸入數據。

axis無或整數或整數元組,可選

選擇形狀中長度為 1 的條目的子集。如果選擇了形狀條目大於 1 的軸,則會引發錯誤。

返回

squeezedndarray

輸入數組,但刪除了長度為 1 的所有維度或維度的子集。這始終是 a 本身或對 a 的視圖。請注意,如果所有軸都被擠壓,則結果是 0d 數組而不是標量。

拋出

ValueError

如果 axis 不是 None,並且被擠壓的軸的長度不是 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  
Traceback (most recent call last):
...
ValueError: cannot select an axis to squeeze out which has size not equal to one
>>> np.squeeze(x, axis=2).shape  
(1, 3)
>>> x = np.array([[1234]])  
>>> x.shape  
(1, 1)
>>> np.squeeze(x)  
array(1234)  # 0d array
>>> np.squeeze(x).shape  
()
>>> np.squeeze(x)[()]  
1234

相關用法


注:本文由純淨天空篩選整理自dask.org大神的英文原創作品 dask.array.squeeze。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。