定义 若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])

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐