numpy where 的使用

例如有一个矩阵，3行3列，我想取矩阵的第一列为3的数字：

a=numpy.arange(9).reshape(3,3) #产生一个3x3矩阵
a=
array([[0, 1, 2],
       [3, 4, 5],
       [6, 7, 8]])																
a[numpy.where(a[:,0]==0)]    #numpy.where()返回的是坐标，在numpy.where里面对矩阵a做以限制，只要它的第0列为0的行的坐标。
array([[0, 1, 2]])           #可以看到按照上面的限制，只输出了矩阵a的第一行，因为第一行第一个元素为0																										     numpy.where(a[:,0]>3)
(array([2], dtype=int64),)																																			     从numpy.where()的返回值可以看到，它返回的是满足条件的行号。并且以数组显示出来。