作者:我无眼泪1221 | 来源:互联网 | 2023-09-17 10:34
原标题:numpy 返回指定行的指定元素的位置索引
先上代码,主要语句为np.where(b[c]==1),详细解释如下:
import numpy as np
b = np.array([[-2,-3,0,0,0,6,4,1],[88,1,0,0,0,6,4,2],[99,6,0,0文章来源站点https://www.yii666.com/,1,6,4,2]]) # 三行八列的数组b
print('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的位置索引
观察np.where()的返回值(array([0, 1], dtype=int64), array([4, 7], dtype=int64))可以发现,
www.yii666.com对于数组b[c]来文章来源地址41514.html说,其中值为1的位置在[0,4]以及[1,7]上,np.where()返回了一个元组,[0]的位置表示行索引,[1]表示列索引。
综上,np.where(b[c]==1)实现的功能是,从b中www.yii666.com按c中元素的顺序找到c所指定的行,然后再这些行中发现值为1的位文章来源地址41514.html置。
来源于:numpy 返回指定行的指定元素的位置索引