Построение отдельных графиков рассеяния с использованием MatPlotLib

Question

У меня есть фрейм данных, и я пытаюсь построить отдельные диаграммы рассеяния для каждого кода группы организации, используя MatPlotLib. Я получаю выходной график объединяет и перекрывает друг друга.

Вот мой код построения

import matplotlib.pyplot as plt

PubProtect = SCPlot[SCPlot['Organization Group Code'] == 1]
x1 = PubProtect['Total Salary']
y1 = PubProtect['Total Benefits']
area1 = 5
plt.scatter(x1, y1, area1, alpha=0.5)

PubWorks = SCPlot[SCPlot['Organization Group Code'] == 2]
x2 = PubWorks['Total Salary']
y2 = PubWorks['Total Benefits']
colors = PubWorks[['Total Salary']].count()
area2 = 5
plt.scatter(x2, y2, area2, alpha=0.5)

Как разделить оба графика?

Answer 1

Возможно, вы захотите проверить Pandas функциональные возможности построения графиков, например df.plot.scatter

В этом случае вы можете сгруппировать свой DataFrame по коду группы организации и сделать разброс график для каждого кода.

grouped = SCPlot.groupby('Organization Group Code')
grouped.plot.scatter('Total Salary', 'Total Benefits')

Это создаст отдельный график для каждого кода группы

Answer 2

Вы можете настроить фигуру и ось следующим образом:

fig, ax = plt.subplots(nrows=2, ncols=1)
ax[0].scatter(x1, y1, area1, alpha=0.5)
ax[1].scatter(x2, y2, area2, alpha=0.5)