目的:在numpy数组中知道指定元素的索引
函数: np.argwhere
>>>x
>>>array([[0, 1, 2],
[3, 4, 5]])
>>>np.argwhere(x>1)
>>>array([[0, 2],
[1, 0],
[1, 1],
[1, 2]])
目的:在numpy数组中知道指定元素的索引
函数: np.argwhere
>>>x
>>>array([[0, 1, 2],
[3, 4, 5]])
>>>np.argwhere(x>1)
>>>array([[0, 2],
[1, 0],
[1, 1],
[1, 2]])