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
,截断了,所以只返回第二行的数组切片,其中包含两个元素。
码字不易,如果大家觉得有用,请高抬贵手给一个赞让我上推荐让更多的人看到吧~