np.nonzero &np.count_nonzero

import numpy as np
test = np.array([[1,2,3],[2,4,5]])
print(np.transpose(np.nonzero(test)).shape[0])
print(np.count_nonzero(test))

输出

6
6

解释：
np.nonzero以元组形式返回array中非0的坐标
np.count_nonzero返回array中非0项的个数
利用np.transpose(np.nonzero(test)).shape[0]可以达到与np.count_nonzero相同的效果