当前位置: 首页 > 工具软件 > compress > 使用案例 >

numpy.compress()函数

南门新荣
2023-12-01

numpy.compress()调用方法:

numpy.compress(condition, a, axis=None, out=None)

各个参数意义:
condition:一维布尔值数组。
a:类数组对象,需要抽取返回值的数组对象。
axis:从哪个坐标轴取元素
out:输出数组,(不推荐设置,否则可能会报错),推荐使用返回值获取最终的数组
返回值:返回满足条件的元素。
代码如下:

import numpy as np


class Debug:
    @staticmethod
    def mainProgram():
        array = np.array([[1, 2],
                          [3, 4],
                          [5, 6]])

        array1 = np.compress([0, 1, 1], array, axis=0)
        print('array1的值为: ')
        print(array1)


if __name__ == "__main__":
    main = Debug()
    main.mainProgram()
"""
array1的值为: 
[[3 4]
 [5 6]]
"""

我们可以看到,axis=0表示沿着y轴选取元素或者数组切片,这时我们设置的布尔数组是[0, 1, 1],所以此时只返回第一行和第二行的元素,即沿着y轴的第一个和第二个数组切片。对于这里为什么axis=0代表沿着y轴,可以查看numpy数组的坐标轴问题(超链接点击跳转)。

再看一个例子:

import numpy as np


class Debug:
    @staticmethod
    def mainProgram():
        array = np.array([[1, 2],
                          [3, 4],
                          [5, 6]])

        array1 = np.compress([0, 1], array, axis=0)
        print('array1的值为: ')
        print(array1)


if __name__ == "__main__":
    main = Debug()
    main.mainProgram()
"""
array1的值为: 
[[3 4]]
"""

我们可以看到。沿着y轴选取数组切片,但是此时的布尔数组只有两个值,因此默认第三个值为False,截断了,所以只返回第二行的数组切片,其中包含两个元素。

码字不易,如果大家觉得有用,请高抬贵手给一个赞让我上推荐让更多的人看到吧~

 类似资料: