Apply Custom Classes / Functions on Pandas DataFrames

Question

I have a class DataImporter with a method called getData that I want to apply on a pandas dataframe. The problem: the class / method can just handle single elements.

Imagine I have a DataFrame with three columns id, a and b.

What I actually want to do is something like: (pseudo code)

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

I found out there is something like pandas.Series.apply, but I don't see that it works for the getData part.

jezrael · Accepted Answer · 2017-06-02 13:21:43Z

2

I think you need apply with axis=1 for process data by scalars in columns:

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

edited Jun 2, 2017 at 13:21

answered Jun 2, 2017 at 13:06

jezrael

868k103 gold badges1.4k silver badges1.3k bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Apply Custom Classes / Functions on Pandas DataFrames

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related