完成函数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])