結果 : how to create square matrix in python