結果 : square matrix python