统计数组或张量中全零行向量的个数

mtx = np.arange(15).reshape(5,3)
mtx[2,:] = np.zeros((1,3))
print(mtx)
fr = np.count_nonzero(mtx,1)
sec = np.not_equal(fr, 0)
print(fr)
print(sec)
print(np.sum(sec))

对于三维或者更高维度的数组，根据想要统计的情况，调整fr = np.count_nonzero(mtx,1) 中的维度即可。

标签：nonzero,mtx,中全,张量,fr,print,行向量,sec,np
来源： https://blog.csdn.net/qq_38081870/article/details/118270584