获取numpy的第一个非0元素索引

import numpy as np
arr=np.array([0,0,2,3,0,4])
print((arr!=0).argmax(axis=0))
#2

扫码关注

实用AI客栈

获取最新AI资讯与实战案例

小编微信号 : langu86

福利区:独学而无友,则孤陋而寡闻.加入机器学习与深度学习讨论QQ群(581789266),一块交流与探讨,共同成长和进步!