Python 使用/不使用色条调整图形的形状

我有一个代码来画一个图形。 当我运行这段代码而不添加plt.colorbar时，我可以得到一个看起来更像矩形的图形。但是，如果我添加颜色条，形状会变为正方形

如何添加颜色条并保持图形的原始形状？谢谢

#%%
import numpy as np
import matplotlib.pyplot as plt
fig = plt.figure()

x = np.random.rand(10000)
y = np.random.rand(10000)
plt.scatter(x,y,c=y)
#plt.colorbar()

plt.show()

接下来，您需要向轴添加一些设置。如果我在创建fig后插入这些行，您的脚本将以正确的方式为我工作：

---

颜色条从轴上窃取空间。如果您希望轴更宽，只需使图形更宽即可。

ax = plt.gca()
ax.set_aspect('equal', 'box')