Plotten von Regressionsgeraden und Datenpunkten
Verfasst: Samstag 10. April 2021, 09:31
Hallo allerseits,
ist es möglich Punkte von einem Pandas Dataframe und eine Regressiongerade auf demselben Graphen zu zeichnen?
Aktuell habe ich ein Dataframe welches die Punkte schön in einem Scatterplot darstellt. Ich bekomme die Regressionsgerade aber irgendwie nicht auf den Plot.
ist es möglich Punkte von einem Pandas Dataframe und eine Regressiongerade auf demselben Graphen zu zeichnen?
Aktuell habe ich ein Dataframe welches die Punkte schön in einem Scatterplot darstellt. Ich bekomme die Regressionsgerade aber irgendwie nicht auf den Plot.
Code: Alles auswählen
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
ausbildungsdauer = [9, 13, 15, 18, 20]
jahresgehalt = [18, 37, 61, 125, 59]
df = pd.DataFrame({'x': ausbildungsdauer, 'y': jahresgehalt})
df.plot('x', 'y', kind="scatter")
print("Der Korrelationskoeffizient beträgt: " + str(df.corr().iloc[0][1]))
m, b = np.polyfit(ausbildungsdauer, jahresgehalt, 1)
plt.plot(x, m*x + b)
plt.show()
print("m: " + str(m))
print("b: " + str(b))