結果 : make a plot square python