Отрегулируйте форму фигур с / без цветовой шкалы

Question

У меня есть код для построения фигуры. Когда я запускаю этот код без добавления 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()

Answer 1

После этой документации вам необходимо добавить некоторые настройки для осей. Ваш скрипт работает для меня правильно, если я вставлю эти строки после создания fig:

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