怎么用python的if 判断矩阵一列只有一个非0的数

假设有一个矩阵 A，可以使用以下代码判断矩阵 A 的第一列是否只有一个非零元素：

import numpy as np

A = np.array([[1, 0, 0],
              [0, 2, 0],
              [0, 0, 0]])

if np.count_nonzero(A[:,0]) == 1:
    print("第一列只有一个非零元素。")
else:
    print("第一列有多个或没有非零元素。")

输出结果为：

第一列只有一个非零元素。

解释：

- `np.count_nonzero(A[:,0])` 用于计算矩阵 A 第一列中非零元素的个数。
- 如果非零元素的个数为 1，则说明第一列只有一个非零元素。
- 如果非零元素的个数为 0 或大于 1，则说明第一列有多个或没有非零元素。