溫馨提示×

NumPy數組壓縮的方法是什么

小億
121
2024-05-13 12:47:13
欄目: 編程語言

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的元素。

0