numpy中的nonzero()的用法

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

[python]  view plain  copy
  在CODE上查看代码片 派生到我的代码片
  1. >>> b1=np.array([TrueFalseTrueFalse])  
  2. >>> np.nonzero(b1)  
  3. (array([02], dtype=int64),)  
对于二维数组b2,nonzero(b2)所得到的是一个长度为2的元组。它的第0个元素是数组a中值不为0的元素的第0轴的下标,第1个元素则是第1轴的下标,因此从下面的结果可知b2[0,0]、b[0,2]和b2[1,0]的值不为0:
[python]  view plain  copy
  在CODE上查看代码片 派生到我的代码片
  1. >>> b2 = np.array([[TrueFalseTrue], [TrueFalseFalse]])  
  2. >>> np.nonzero(b2)  
  3. (array([001], dtype=int64), array([020], dtype=int64))  
当布尔数组直接做为 numpy数组 下标时,相当于使用由nonzero()转换之后的元组作为下标对象:
[python]  view plain  copy
  在CODE上查看代码片 派生到我的代码片
  1. >>> a = np.arange(3*4*5).reshape(3,4,5)  
  2. >>> a  
  3. array([[[ 0,  1,  2,  3,  4],  
  4.         [ 5,  6,  7,  8,  9],  
  5.         [1011121314],  
  6.         [1516171819]],  
  7.   
  8.        [[2021222324],  
  9.         [2526272829],  
  10.         [3031323334],  
  11.         [3536373839]],  
  12.   
  13.        [[4041424344],  
  14.         [4546474849],  
  15.         [5051525354],  
  16.         [5556575859]]])  
  17. >>> a[b2]  
  18. array([[ 0,  1,  2,  3,  4],  
  19.        [1011121314],  
  20.        [2021222324]])  
  21. >>> a[np.nonzero(b2)]  
  22. array([[ 0,  1,  2,  3,  4],  
  23.        [1011121314],  
  24.        [2021222324]])  

为了观察变化,下面有一个例子:

[python]  view plain  copy
  在CODE上查看代码片 派生到我的代码片
  1. dataSet=array(  
  2.     [[1,0,0,0],  
  3.     [0,1,0,0],  
  4.     [0,1,0,0],  
  5.     [0,0,0,1]])  
  6. a=dataSet[:,1]>0.5  
  7. print(a)  
  8. print('--------------')  
  9. print(nonzero(a))  
  10. print('--------------')  
  11. print(nonzero(a)[0])  
  12. print('--------------')  
  13. print(dataSet[nonzero(a)[0],:])  
输出结果:

[python]  view plain  copy
  在CODE上查看代码片 派生到我的代码片
  1. [False  True  True False]  
  2. --------------  
  3. (array([12], dtype=int64),)  
  4. --------------  
  5. [1 2]  
  6. --------------  
  7. [[0 1 0 0]  
  8.  [0 1 0 0]]  

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值