как организовать линейную линию в соответствии с точками данных

Question

У меня есть набор данных 2D. Ось Y идет от 5 до -200, а ось X - от 0 до 100. У меня есть значения наклона и пересечения, чтобы нарисовать линейную линию между точками данных. Однако у меня есть некоторые проблемы со зрением.

Вот как выглядят мои данные при их построении:

Вот как выглядит мой график, когда я рисую линию.

Способ, которым я строю свою линию:

axis.set_xlim([0,101])
x_vals = np.array(axis.get_xlim())
y_vals = intercept + slope * x_vals
plt.plot(x_vals, y_vals, c="black")

Я хотел бы сохранить первое изображение и добавить линию поверх него. Мне было интересно, может ли кто-нибудь помочь мне.

Answer 1

Я не вижу ничего плохого. Это просто изменения y-пределов из-за новой строки. Чтобы исправить это, сбросьте пределы y:

axis.set_xlim([0,101])
x_vals = np.array(axis.get_xlim())

# save the old y limits
y_lims = axis.get_ylim()

# these new y_vals are larger than the old y_lims
y_vals = intercept + slope * x_vals
axis.plot(x_vals, y_vals, c="black")

# reset y_limits
axis.set_ylims(y_lims)