np.where与np.argwhere共同点与区别分析

1.问题描述
我本来想用np.where去找3维array的二维平面中指定数字2对应的depth维度的值，结果没有搞出来。上网搜索了一下，发现还有一个np.argwhere，解决了问题。

2.np.where与np.argwhere的共同点与区别
① 共同点
都是用来找出满足指定statement的数的index，所谓的statement就是np.where(statement)中的条件表达式，如：x==2

② 不同点
a = np.where(x==2)的结果是tuple类型的，但是你千万别以为是类似于(0,0)这样简单的，实际是这样蛋疼的格式：（array([0], dtype=int64),  array([0], dtype=int64))， 其实乍一看和(0,0)表达的意思是一样的，但是如果用这个来索引其他的array就会有点问题，当然了，x(a)得到的肯定全是数字2。

（array([0], dtype=int64), array([0], dtype=int64))这样的结构很烦，那怎么搞成简单的(0,0)呢？那就要请出np.argwhere。a = np.argwhere(x==2) 的输出是这样的：array([0,0], dtype=int64)，这样是不是就舒服多了，但是x(a)是会报错的！那怎么办呢？我们考虑使用 tuple(array([0,0], dtype=int64))将array转成( )tuple类型，结果就是(0,0)了。

3.本人现在的研究方向是：
图像的语义分割，如果有志同道合的朋友，可以组队学习
haiyangpengai@gmail.com
q 1355365561