在熊猫数据框上应用自定义类/函数

Question

我有一个DataImporter类，我想要应用在熊猫数据框上的方法叫做getData。问题：类/方法只能处理单个元素。

想象一下，我有一个DataFrame有三列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)