Python numpy compress用法及代碼示例

本文簡要介紹 python 語言中 numpy.compress 的用法。

用法:

numpy.compress(condition, a, axis=None, out=None)

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

沿給定軸工作時，沿該軸的切片返回輸出對於每個索引健康)狀況評估為真。在處理一維數組時，compress相當於numpy.extract.

參數：

condition： 一維布爾數組

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

a： array_like

從中提取零件的數組。

axis： 整數，可選

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

out： ndarray，可選

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

返回：

compressed_array： ndarray

a 的副本，沒有沿軸的切片，條件為假。

例子：

>>> 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])