# Import NumPy and matplotlib.pyplot
import numpy as np
import matplotlib.pyplot as plt
# Read in the data for age, bmi and insurance charges in separate arrays.
# Print out to check
#
filename='insurance.csv'
age = np.loadtxt(filename,usecols=0,skiprows=1,delimiter=',',dtype=int)
bmi = np.loadtxt(filename,usecols=2,skiprows=1,delimiter=',')
charges = np.loadtxt(filename,usecols=6,skiprows=1,delimiter=',')
print (age)
print (bmi)
print (charges)
[19 18 28 ... 18 21 61]
[27.9 33.77 33. ... 36.85 25.8 29.07]
[16884.924 1725.5523 4449.462 ... 1629.8335 2007.945 29141.3603]
# print out the number of data instances
print (f'There are {len(age)} data instances.')
There are 1338 data instances.
# plot charges vs age and add labels to axis and title
plt.scatter(age,charges)
plt.xlabel('Age')
plt.ylabel("Charges")
plt.title("Age vs. Charges")
# plot charges vs BMI; add axis labels and plot title
plt.scatter(age,bmi)
plt.xlabel('Age')
plt.ylabel("BMI")
plt.title("Age vs. BMI")
# Import NumPy and matplotlib.pyplot
import numpy as np
import matplotlib.pyplot as plt
# Read in the data for bmi and insurance charges in separate arrays.
# Print out to check
#
filename='insurance.csv'
bmi = np.loadtxt(filename,usecols=2,skiprows=1,delimiter=',')
charges = np.loadtxt(filename,usecols=6,skiprows=1,delimiter=',')
print (bmi)
print (charges)
[27.9 33.77 33. ... 36.85 25.8 29.07]
[16884.924 1725.5523 4449.462 ... 1629.8335 2007.945 29141.3603]
# Fit a line using linear regression
line_coeff = np.polyfit(bmi,charges,1)
print (f'The line equation is {line_coeff[0]}x + {line_coeff[1]}')
The line equation is 393.87303079739513x + 1192.937208961152
# Plot data and line on same graph
min_bmi = min(bmi)
max_bmi = max(bmi)
x = np.linspace(min_bmi,max_bmi,100)
f = np.poly1d(line_coeff)
plt.scatter(bmi,charges)
plt.plot(x,f(x),'r')
plt.xlabel("BMI")
plt.ylabel("Charges")
plt.title("BMI vs. Charges")
# Fit a parabola to data
poly_coeff = np.polyfit(bmi,charges,2)
print (f'{poly_coeff[0]}x^2 + {poly_coeff[1]}x + {poly_coeff[2]}')
-6.662322630063166x^2 + 813.9749575731079x + -5177.034125706157
# Plot data and parabola on same graph
min_bmi = min(bmi)
max_bmi = max(bmi)
x = np.linspace(min_bmi,max_bmi,100)
f = np.poly1d(poly_coeff)
plt.scatter(bmi,charges)
plt.plot(x,f(x),'r')
plt.xlabel("BMI")
plt.ylabel("Charges")
plt.title("BMI vs. Charges")