import pandas as pd
import matplotlib.pyplot as plt

# turn on print for all rows
pd.set_option('display.max_rows', None)
us_data = 'US-Data.csv'

master_us_df = pd.read_csv(us_data)

column_filter = ['Age','Centimeters']

us_male_filter = master_us_df['Gender'] == 'male'
filtered_us_male_data = master_us_df.loc[us_male_filter,column_filter]
grouped_us_male_data = filtered_us_male_data.groupby(['Age']).mean()

us_female_filter = master_us_df['Gender'] == 'female'
filtered_us_female_data = master_us_df.loc[us_female_filter,column_filter]
grouped_us_female_data = filtered_us_female_data.groupby(['Age']).mean()

master_us_data = None
filtered_us_male_data = None
filtered_us_female_data = None

axes = plt.axes()

plt.plot(grouped_us_male_data.index,grouped_us_male_data['Centimeters'], color='blue', label="Male")
plt.plot(grouped_us_female_data.index,grouped_us_female_data['Centimeters'], color='green', label="Female")

plt.title("Height vs Age for US Records")
plt.xlabel("Age")
plt.ylabel("Height (in Centimeters)")

axes.legend(loc='best', frameon=False)

plt.grid()
plt.show()