結果 : how to make a square matrix in python