def plot_survival_by_age_category():
train_data = di.get_titanic_data()
train_data = di.filter_age(train_data)
age_categories_pivot = train_data.pivot_table(index="Age_category",
values="Survived")
age_categories_pivot.plot.bar()
plt.title("Graph to Show Survival Rate by Age Category")
plt.xlabel("Age Group")
plt.ylabel("Survival Rate")
plt.savefig("Graph to Show Survival Rate by Age Category")
