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]])
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]])