带有组的python中的散点图

Question

下面是三列VMDensity，ServerswithCorrectable错误和VMReboots。

VMDensity   correctableCount    avgVMReboots
LowDensity  7   5
HighDensity 1   23
LowDensity  5   11
HighDensity 1   23
LowDensity  9   5
HighDensity 1   22
HighDensity 1   22
LowDensity  9   2
LowDensity  9   6
LowDensity  5   3

我尝试了以下操作，但不确定如何通过不同颜色的组来创建它。

import matplotlib.pyplot as plt
import pandas as pd

plt.scatter(df.correctableCount, df.avgVMReboots)

现在，我需要根据VMDensity的分组生成一个scatter plot。低密度虚拟机应使用一种颜色，高密度虚拟机应使用另一种颜色。

Answer 1

如果我对您的理解正确，则无需将数据“分组”：无需考虑绘制所有数据点。您只想为它们着色不同。因此，尝试类似

plt.scatter(df.correctableCount, df.avgVMReboots, c=df.VMDensity)

您将需要将df.VMDensity字符串映射到数字和/或使用scatter的cmap参数进行播放。

请参阅此example from matplotlib's gallery。