我有一个类 DataImporter
,其中包含一个名为 getData
的方法,我想将其应用于 pandas 数据框。问题:类/方法只能处理单个元素。
假设我有一个包含三列 id
、a
和 b
的 DataFrame。
我真正想做的是:(伪代码)
df["c"] = Class(df["id"]).getData(df["a"], df["b"])
我发现有类似 pandas.Series.apply 的东西,但我没有看到它适用于 getData
部分。
最佳答案
我认为你需要apply
使用 axis=1
按列中的标量处理数据:
df["c"] = df.apply(lambda x: Class(x["id"]).getData(x["a"], x["b"]), axis=1)
关于python - 在 Pandas DataFrame 上应用自定义类/函数,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/44329698/