定义 若a是一个numpy数组,和它shape相同bool型的numpy数组称为a的一个mask
a[mask]输出的是一个一维数组,它是由满足如下的元素构成
mask为True的位置在a中对应的元素
举个例子
import numpy as np
a=np.array([[1,2],[3,4]])
mask=a>1
print(a[mask])
结果为: array([2, 3, 4])
定义 若a是一个numpy数组,和它shape相同bool型的numpy数组称为a的一个mask
a[mask]输出的是一个一维数组,它是由满足如下的元素构成
mask为True的位置在a中对应的元素
举个例子
import numpy as np
a=np.array([[1,2],[3,4]])
mask=a>1
print(a[mask])
结果为: array([2, 3, 4])