np.compress()函数

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，截断了，所以只返回第二行的数组切片，其中包含两个元素。

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