如何用python科学计算中的矩阵替代循环?