Simple Data Science with Python - Linear Regression

import pandas as pd
from sklearn.linear_model import LinearRegression

import matplotlib.pyplot as plt

data = pd.read_csv("salary.csv")

ages = data.iloc[:, 0:1]

salaries = data.iloc[:, 1:2]

print ages
print salaries

lr = LinearRegression()

lr.fit(ages, salaries)

salaries_prediction = lr.predict(ages)

print salaries_prediction


plt.plot(ages, salaries_prediction)
plt.scatter(ages, salaries_prediction)

plt.show()