目的:在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]])