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


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


用法:

dask.array.compress(condition, a, axis=None)

沿給定軸返回數組的選定切片。

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

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

沿給定軸工作時,沿該軸的切片在 output 中為每個索引返回,其中 condition 的計算結果為 True。在處理一維數組時,compress 等效於 extract

參數

condition一維布爾數組

選擇要返回的條目的數組。如果 len(condition) 小於 a 沿給定軸的大小,則輸出將被截斷為條件數組的長度。

aarray_like

從中提取零件的數組。

axis整數,可選

沿其進行切片的軸。如果無(默認),則處理展平數組。

outndarray,可選(在 Dask 中不支持)

輸出數組。它的類型被保留,它必須具有正確的形狀來保存輸出。

返回

compressed_arrayndarray

a 的副本,沒有沿軸的切片,其中 condition 為假。

例子

>>> a = np.array([[1, 2], [3, 4], [5, 6]])  
>>> a  
array([[1, 2],
       [3, 4],
       [5, 6]])
>>> np.compress([0, 1], a, axis=0)  
array([[3, 4]])
>>> np.compress([False, True, True], a, axis=0)  
array([[3, 4],
       [5, 6]])
>>> np.compress([False, True], a, axis=1)  
array([[2],
       [4],
       [6]])

處理展平數組不會沿軸返回切片,而是選擇元素。

>>> np.compress([False, True], a)  
array([2])

相關用法


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