从数据框中制作多个图的函数

Question

我有一个大型数据集。我试图弄清楚如何编写一个函数，该函数将为每种类型获取 x 和 y 输入并绘制它们。 ...我对 python 很陌生，函数让我很困惑。

例如 - 我有关于狗的数据。我对每种狗类型的身高和体重进行了多次测量......有没有办法编写一个函数，将所有这些测量值绘制在每种狗类型的单独散点图上？做这样的事情：plot 没有手动拆分每种类型的 df 并绘制它（这就是我在这里所做的）.. 我有 60 种不同的类型>

Answer 1

我通常做的是首先获取数据框中唯一类型的列表（假设它的变量名是df）。现在，对于每种类型，我们都会得到 df 的不同部分，我在示例中将其称为 df_tmp：

for t in df["types"].unique():
    df_tmp = df[df["type"] == t]

现在您可以使用df_tmp 中的数据调用绘图函数，这实际上只是整个df 的一个子集。

就性能而言，这可能不是最好的解决方案，但对我来说它非常直观。