python基础--nonzero

nonzero(a) 将对矩阵a的所有非零元素， 分别安装两个维度， 一次返回其在各维度上的目录值。

#coding:utf-8
from numpy import *

a = mat([[1, 0, 0], [0, 2, 0],[0, 0, 0]])
print  nonzero(a)
#结果为：(array([0, 1], dtype=int64), array([0, 1], dtype=int64))
print  nonzero(a)[1]
#结果：返回第二维度上的值，[0 1]

a = mat([[1, 0, 0], [0, 2, 0],[0, 2, 2]])
s= nonzero(a)[1]
print s
#s 结果为[0 1 1 2]

for i in xrange(len(s)):
    if sum(s==i):     #这个函数是怎么使用的？
        print sum(s==i) 
结果为：
1
2
1