np.flatnonzero

numpy.flatnonzero(a)

  • a:输入array
  • 返回:一个数组,该数组是a的非0索引

该函数输入一个矩阵,返回array的非0索引(也可以指定索引元素,下面会展示)

Return indices that are non-zero in the flattened version of a.

import numpy as np

x = np.arange(-2 ,3)
print(x)
y = np.flatnonzero(x)
print(y)
y_ = x[np.flatnonzero(x)]
print(y_)
'''
[-2 -1  0  1  2]
[0 1 3 4]
[-2 -1  1  2]
'''


# also can use like this
xx = np.array([1, 3, 2, 3, 4, 3, 1, 2])
yy = np.flatnonzero(xx == 3)
print(yy)
'''
[1 3 5]
'''
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值