結果 : how to create a square matrix in python