在Pandas DataFrames上应用自定义类/函数

Question

我有一个类DataImporter，其中包含一个名为getData的方法，我希望将其应用于pandas数据帧。问题是：类/方法只能处理单个元素。

想象一下，我有一个包含三列id，a和b的数据框。

我真正想做的是:(伪代码）

df["c"] = Class(df["id"]).getData(df["a"], df["b"])

我发现有类似pandas.Series.apply的内容，但我认为它不适用于getData部分。

Answer 1

我认为您需要apply axis=1以及列中的标量来处理流程数据：

df["c"] = df.apply(lambda x: Class(x["id"]).getData(x["a"], x["b"]), axis=1)