2021
06-22
06-22
numpy 实现返回指定行的指定元素的位置索引
先上代码,主要语句为np.where(b[c]==1),详细解释如下:importnumpyasnpb=np.array([[-2,-3,0,0,0,6,4,1],[88,1,0,0,0,6,4,2],[99,6,0,0,1,6,4,2]])#三行八列的数组bprint('b\n',b)c=np.array([2,0])#c表示指定行print('b[c]\n',b[c])#b[c]返回数组b的指定行这里依次返回了b的下标为2和0的行print('\n')print(np.where(b[c]==1))#返回指定行的指定元素的位置索引这里返回了b[c]每行中值为1的位置索...
继续阅读 >