在 scipy 中使用共轭梯度法求解线性方程组

如何解决在 scipy 中使用共轭梯度法求解线性方程组

我想使用 scipy.sparse.linalg.cg 求解格式为 Ax=b 的方程组。 我使用以下代码行

from scipy.sparse.linalg import cg
x=cg(arg_stiffness,arg_force,1e-10,300 )

arg_stifness 和 arg_force 是 csr_matrix 格式的形状为 (593457,593457) 和 (593457,1) 的稀疏数组，容差为 1e-10，最大迭代次数为 300。

当我运行此代码时，出现以下错误：

ValueError: A and b have incompatible dimensions

有人可以帮我解决这个问题吗？

非常感谢！