提取矩阵每一行非零下标及个数

 

import numpy as np

 这里也可以导入表格数据

data = [[1,0,1,0,0],[0,4,0,0,4],[0,0,2,3,0]]
print(data)

output:[[1, 0, 1, 0, 0], [0, 4, 0, 0, 4], [0, 0, 2, 3, 0]]

np.nonzero这个方法可以把矩阵非零数的行、列下标提取出来

item_zero = np.nonzero(data)
print(item_zero)

output:(array([0, 0, 1, 1, 2, 2], dtype=int64), array([0, 2, 1, 4, 2, 3], dtype=int64))

将行、列下标数据分开，用zip()函数一一对应后组成数组

item_zero = list(zip(item_zero[0], item_zero[1]))
print(item_zero)

output:[(0, 0), (0, 2), (1, 1), (1, 4), (2, 2), (2, 3)]

得到每一行非零数的个数：

d = {}
for i in item_zero:
    d[i[0]] = d.get(i[0],0) + 1
print(d)

output:{0: 2, 1: 2, 2: 2}