使用高斯-若尔当消元法的思想对方程求解

算是在学习线性代数的路程中小小的记录一下吧，用code把思想转化为实际运用，感觉对于理解也许有很好的帮助。

 在代码之中的参数主要是x和y，x为公式系数组成的矩阵，需要保证这个公式的位置在列上面以及进行了未知数的对齐，比如第一列对应的未知数是a，第二列对应的未知数为b。y则是公式的里面的常数。

比如x矩阵有一行为1， 1， 1而y对应的值为6，表示为：

a+b+c = 6

def question(x, y):
  w = [0] * len(x)
  # 消元
  for i in range(len(x)):
    if findMainX(x, y, i):
      for j in range(i+1, len(x)):
        g = x[j][i]
        y[j] -= float(y[i] * g)/x[i][i]
        for k in range(i, len(x)):
          x[j][k] = x[j][k] - float(x[i][k]*g)/x[i][i]
    else: return [-1]*len(w)

  # 归元
  for i in range(len(w)):
    for j in range(len(w)-1, len(w)-1-i, -1):
      y[len(w)-1-i] -= x[len(w)-1-i][j] * w[j]
    w[len(w)-1-i] = float(y[len(w)-1-i]) / x[len(w)-1-i][len(w)-1-i]
  return w

def findMainX(x, y, i):
  if x[i][i] != 0: return True
  for j in range(i+1, len(x)):
    if x[j][i] != 0:
      x[i], x[j] = x[j], x[i]
      y[i], y[j] = y[j], y[i]
      return True
  return False

if __name__ == '__main__':
  # 设解为 1 2 3
  x = [[1, 1, 1], [5, 3, -10], [-2, -6, 7]]
  y = [6, -19, 7]
  print(question(x, y))

 在代码之中，主要是进行消元和归元的操作，消元，自上而下的去确定主元，这里我用到了一个辅助的函数，当下一行的主元位置为0时，则去找一行在主元位置不为零的进行置换，如果找不到，说明维度不够，无解。

归元主要是使用循环的一个逆向，来从最末尾的一个主元（表现的形式为，一个未知数.*系数对于一个常数，可求得当前未知数的解），向前求解，则最终取得答案。