Python NumPy compress方法用法及代碼示例

例子

壓縮一維數組

np.compress([True, False, True, False], [4,5,6,7])



array([4, 6])

請注意，這相當於以下內容：

a = np.array([4,5,6,7])
mask = [True, False, True, False]
a[mask]



array([4, 6])

截斷

當條件數組的大小小於輸入數組的大小時，就會發生截斷：

a = np.array([4,5,6,7])
np.compress([True, False], a)



array([4])

這裏，僅考慮數組中的前兩個值 - 但僅返回第一個值，因為第二個值被標記為 False。

壓縮二維數組

考慮以下二維數組：

a = np.array([[4,5],[6,7]])
a



array([[4, 5],
       [6, 7]])

從展平數組中提取

np.compress([True, False, False, True], a)



array([4, 7])

逐行提取

np.compress([True, False], a, axis=0)



array([[4, 5]])

按列提取

np.compress([True, False], a, axis=1)



array([[4],
       [6]])