# -*- coding: utf-8 -*-

import pandas as pd
import matplotlib.pyplot as plt
import numpy as np
from sklearn.linear_model import LinearRegression

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

print(data.columns)

expenses = data.expenses.values.reshape(-1,1)
ageBmis = data.iloc[:,[0,2]].values
regression = LinearRegression
regression.fit(ageBmis,expenses)
print(regression.predict(np.array([[20,20]])))