Add columns to extended `pandas.DataFrame` class

Question

I created a subclass of pandas.DataFrame to bundle a few functionalities:

class ABT(pandas.DataFrame):
    def __init__(self, data=None, ...):
        if data is None:
            ...
            data = DataFrame(..., tz='utc'))
        super(ABT, self).__init__(data)

I want to make a method that uses another DataFrame as a parameter and appends it to ABT. The question is: How to join/merge/concat to self?

    def add_df(self, new_df):
        df_utc = new_df.tz_localize('CET', ambiguous='NaT').tz_convert('utc')

        ...

        self.merge(df_utc , how='left', inplace=True)

The above method does not work, but I hope there is a ncie way to solve this.

Stef · Accepted Answer · 2019-08-13 09:41:09Z

1

There is no inplace keyword for merge. Besides of this, you should read Subclassing pandas datastructures. So you need something like this:

class ABT(pd.DataFrame):

    @property
    def _constructor(self):
        return ABT

    @property
    def _constructor_sliced(self):
        return pd.Series

    def add_df(self, new_df):
        return self.merge(new_df, how='left')



abt1 = ABT(df1)
abt2 = ABT(df2)
abt1 = abt1.add_df(abt2)

If you don't define the constructors, the result of add_df() will be of type pandas.core.frame.DataFrame instead of ABT.

edited Aug 13, 2019 at 9:41

answered Aug 13, 2019 at 9:28

Stef

30.9k3 gold badges34 silver badges60 bronze badges

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

Collectives™ on Stack Overflow

Add columns to extended `pandas.DataFrame` class

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