在Python中构建增广矩阵的实现方法

# TODO 构造增广矩阵，假设A，b行数相同def augmentMatrix(A, b):  if(len(A) != len(b)):    raise 'The number of rows is different'  result = []  for i in range(len(A)):    row = []    for j in range(len(A[i])):      row.append(A[i][j])    for j in range(len(b[i])):      row.append(b[i][j])    result.append(row)      return result

# TODO 构造增广矩阵，假设A，b行数相同def augmentMatrix(A, b):  return [AA + bb for AA, bb in zip(A,b)] A = [[1,2,3],[4,5,6],[7,8,9]]b = [[1],[2],[3]]print augmentMatrix(A,b)[[1, 2, 3, 1], [4, 5, 6, 2], [7, 8, 9, 3]]