Python中多行中非零值的计数？

import numpy as np

# 创建一个多行多列的二维数组
arr = np.array([[1, 0, 3, 0],
                [0, 2, 0, 4],
                [5, 0, 0, 6]])

# 计算每行非零值的数量
nonzero_counts = np.count_nonzero(arr, axis=1)

# 打印结果
for i, count in enumerate(nonzero_counts):
    print("第{}行非零值的数量为：{}".format(i+1, count))

第1行非零值的数量为：2
第2行非零值的数量为：2
第3行非零值的数量为：3