学堂 学堂 学堂公众号手机端

NumPy数组压缩的方法是什么

lewis 3年前 (2022-01-19) 阅读数 5 #技术

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


版权声明

本文仅代表作者观点,不代表博信信息网立场。

热门