python计算矩阵主元列

这是一个以 Python 计算矩阵主元列的示例代码：

import numpy as np

# 定义一个矩阵
matrix = np.array([
    [1, 2, 3],
    [0, 4, 5],
    [0, 0, 6]
])

# 计算矩阵的主元列
pivot_cols = np.argwhere(matrix.any(axis=0)).flatten()

print("矩阵的主元列为：")
print(pivot_cols)

上述代码中，我们使用了 NumPy 库来实现计算矩阵的主元列。具体步骤包括：

1. 定义一个矩阵；
2. 使用 `any(axis=0)` 方法来获取矩阵中非零元素所在的列；
3. 使用 `argwhere()` 方法和 `flatten()` 方法来获取主元列的位置，并打印输出结果。

希望对你有所帮助。