当前位置:主页 > 软件编程 > Python代码 >

numpy:找到指定元素的索引示例

时间:2021-05-05 15:36:27 | 栏目:Python代码 | 点击:

目的:在numpy数组中知道指定元素的索引

函数: np.argwhere

>>>x
>>>array([[0, 1, 2],
  [3, 4, 5]])
  
>>>np.argwhere(x>1)
>>>array([[0, 2],
  [1, 0],
  [1, 1],
  [1, 2]])

您可能感兴趣的文章:

相关文章