NumPy數組壓縮的方法是使用numpy.compress()
函數。該函數接受兩個參數,第一個參數是一個布爾數組,用于指示哪些元素需要被壓縮,第二個參數是要進行壓縮的數組。壓縮后的數組將只包含布爾數組中為True的元素。例如:
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
condition = np.array([True, False, True, False, True])
compressed_arr = np.compress(condition, arr)
print(compressed_arr)
輸出結果將是[1 3 5]
,只包含布爾數組中為True的元素。