numpy where 的使用

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

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