今天小编就为大家分享一篇在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 fo<b style="color:transparent">来源gao@!dai!ma.com搞$$代^@码!网</b>r 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]]
注:解读一下AA+bb, 在 python中, [1, 2, 3] + [4]这样的表达式会返回[1, 2, 3, 4]
以上就是在Python中构建增广矩阵的实现方法的详细内容,更多请关注gaodaima搞代码网其它相关文章!