使用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

您可以将图形和轴设置为：

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)

Answer 2

您可能想要签出功能，例如df.plot.scatter

在这种情况下，您可以按组织组代码对数据帧进行分组，并为每个代码绘制散点图

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

这将为每个组码生成一个单独的图形