numpy中nonzero()的用法

转自：python numpy中nonzero(),isnan()用法

nonzero(a)---返回数组a中值不为零的元素de下标,,返回值为一个长度为a.ndim(数组a的秩)的元组,元组的每个元素都是一个整数数组,其值为非零元素的下标在对应轴上的值.例如一维布尔数组b1,nonzero(b1)所得到的是长度为1的元组,表示b1[0]和b1[2]的值不为0(False).

注:使用布尔数组直接作为下标对象或者元组下标对象时,相当于使用nonzero()将布尔数组转换成一组整数数组,然后使用整数数组进行下标运算.

 

对于二维数组b2，nonzero(b2)所得到的是一个长度为2的元组。它的第0个元素是数组a中值不为0的元素的第0轴的下标，第1个元素则是第1轴的下标，因此从下面的结果可知b2[0,0]、b[0,2]和b2[1,0]的值不为0：

对于二维数组b2，nonzero(b2)所得到的是一个长度为2的元组。它的第0个元素是数组a中值不为0的元素的第0轴的下标，第1个元素则是第1轴的下标，因此从下面的结果可知b2[0,0]、b[0,2]和b2[1,0]的值不为0