python - 在 Pandas DataFrame 上应用自定义类/函数

Question

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

假设我有一个包含三列 id、a 和 b 的 DataFrame。

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

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)