python 的 矩阵运算

救救孩子吧，想了半天就对了两个测试点，怎么做呀，求代码