2021
07-01
07-01
如何获取numpy的第一个非0元素索引
我就废话不多说了,大家还是直接看代码吧~importnumpyasnparr=np.array([0,0,2,3,0,4])print((arr!=0).argmax(axis=0))#2补充:python获取二维矩阵的每一行的第一个非零元素直接上代码~"""核心函数:array_1D!=0:返回一个True/False序列array.argmax(axis=0):返回沿axis轴的最大元素的索引,当存在多个相等的最大值时,返回第一个最大值的索引"""importnumpyasnparray_1D=np.array([0,1,0,-1,0])array_2D=np.array(...
继续阅读 >