1、nonzero
对于一维数据来说，将返回符合条件的 下标
>>> b1 = np.array([True, False, True, False])
>>> np.nonzero(b1)
(array([0, 2]),)
对于二维数据来说，将返回两维 元组， 第一维是符合条件的 x的索引，第二维是符合条件的y的索引
>>> b2 = np.array([[True, False, True], [True, False, False]])
>>> np.nonzero(b2)
(array([0, 0, 1]), array([0, 2, 0]))
原文：https://www.cnblogs.com/zhaopAC/p/8997706.html