Генерация отдельных графиков координат по столбцу ID в DataFrame с помощью matplotlib

Question

У меня есть DataFrame с идентификаторами и значениями x и y. Мне нужно построить координаты x и y для каждого идентификатора на отдельном графике.

   df = pd.DataFrame({'ID':['A','B','B','A','C','C'], 'X': [5,3,4,2,5,3], 'Y':[1,2,6,4,5,2]})

Для данных этого примера ожидаемый результат будет 3 графиками (A, B и C) каждый с две точки нанесены. Код, который я попробовал, выглядит следующим образом:

   groups = df.groupby('ID')
   fig, ax = plt.subplots()
   for ID,group in groups:
        ax.plot(group.X,group.Y)

Это разделяет точки по идентификатору, но отображает все на одном графике. Как я могу генерировать отдельные графики для каждого идентификатора?

Answer 1

Это потому, что вы строите все в ax. Делай вместо:

groups = df.groupby('ID')

# this is not needed
# fig, ax = plt.subplots()

for ID,group in groups:
    group.plot(x='X',y='Y')

Answer 2

Вы можете напрямую с помощью

df.groupby('ID').plot(x='X', y='Y')

это создаст график для идентификатора