Необходимо понимать, как разбить данные на две отдельные строки на одном графике

Question

У меня есть данные, которые выглядят как выше.

Я успешно вывел общую сумму самоубийств / 100 000 человек за эти годы с помощью следующего кода:

df2 = df2[df2.year != 2016]

fig, ax = plt.subplots(figsize=(15,7))
df2.groupby(['year'])['suicides/100k pop'].sum().plot(marker='o')

Но Мне нужен график, который также показывает 2 разные линии, одну для мужчины и одну для женщины, за один и тот же период времени. Как я могу это сделать?

Answer 1

Подберите данные о сексе перед применением группы, затем постройте график для каждого пола. Вы также можете изменить маркер (или любой другой параметр) во время итерации.

df2 = df2[df2.year != 2016]

fig, ax = plt.subplots(figsize=(15,7))

genders = ('male', 'female')
markers = ('o', 'x')
colors = ('b', 'r')

for b, m, g in zip(colors, markers, genders):
    df2[df2.sex.eq(g)].groupby(['year'])['suicides/100k pop'].sum().plot(marker=m, ax=ax)
ax.legend(genders)