結果 : create a square matrix in python