python第七章课后习题plus(6)

完成函数fun6，参数v是一个只有0和1的n*m矩阵，并求出其中1的个数

这题找出所有为1的索引，然后使用len函数就可以知道有多少个了
做到后面还知道可以用bincount函数

def fun6(v):
    '''
    Arg
    v:a 5*5 matrix ; return the total number of 1
    eg.[[0,1,1,0,1],
        [0,0,1,0,1],
        [0,1,1,0,0],
        [1,1,1,0,1],
        [1,0,1,0,1]]
    return 14
    '''
    v1=np.array(v)
    index=np.where(v1-1==0)
    return len(v1[index])