python 定位array坐标

python中快速寻找array坐标索引

import numpy as np

a = np.array([[0,0,1,2],
              [1,1,0,0]])

# 找到a内大于0的坐标并返回为array

coords = np.array(np.where(a>0))


# 返回值

array([[0,0,1,1],
       [2,3,0,1]])