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

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

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐