結果 : how to make a square plot in python