在熊貓數據框上應用自定義類/函數

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)