means_pre = pre.groupby(["cohort", "treated"], observed=True)["yeduc"].mean()
means_pre = means_pre.unstack("treated")
plt.rcParams.update({"font.size": 15})
fig, ax = plt.subplots(figsize=(8, 4.2), layout="constrained")
for group, colour in [(0, "#0066a1"), (1, "#c43c39")]:
ax.plot(range(4), means_pre[group], marker="o", color=colour,
label=f"Group {group}")
ax.set(xticks=range(4), xticklabels=["-4", "-3", "-2", "-1"],
title="Education across pre-treatment cohorts",
xlabel="Supplied cohort category", ylabel="Mean years of education")
ax.spines[["top", "right"]].set_visible(False)
ax.legend(frameon=False)
plt.show()