python自编程序整理矩阵