Python稀疏矩阵基本运算

python怎么由一个m行n列的稀疏矩阵A创建其三元组表示?

参考下

# 假设稀疏矩阵 A 是一个 m 行 n 列的二维列表
A = [
    [1, 0, 0, 0, 2],
    [3, 0, 0, 0, 0],
    [0, 4, 0, 5, 0],
    [0, 0, 0, 0, 6]
]

# 获取稀疏矩阵的行数和列数
m, n = len(A), len(A[0])

# 创建三元组表示的空列表
triplets = []

# 遍历并将非零元素添加到三元组表示中
for i in range(m):
    for j in range(n):
        if A[i][j] != 0:
            triplets.append((i, j, A[i][j]))

print(triplets)