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")