NumPy数组压缩的方法是什么
lewis
2024-04-21
19次阅读
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的元素。

发表评论:
◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。