how to create a dataframe class in pandas/python

Question

i want to create a class, which will create a dataframe object.

but the below code doesn't work properly:

class Df:

    def __init__(self,df):
        self = df

my_df = Df(df)
print(my_df.columns)

AttributeError: 'Df' object has no attribute 'columns'

what should I do? I want to create custom methods in order to manipulate dataframe, that's why I want to use oop.

Thinking about inheritance?

FavorMylikes
– FavorMylikes

2022-04-01 09:01:30 +00:00
Commented Apr 1, 2022 at 9:01 — FavorMylikes
– FavorMylikes, Commented Apr 1, 2022 at 9:01

Christian Karcher · Accepted Answer · 2022-04-01 09:01:50Z

3

You could directly inherit from Dataframe and add methods to this class:

import pandas as pd

class Df(pd.DataFrame):

    def say_hello(self):
        print('hello world')

df = Df({'data': [1,2,3]})
print(df.columns)
df.say_hello()

answered Apr 1, 2022 at 9:01

Christian Karcher

3,8441 gold badge18 silver badges28 bronze badges

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

Collectives™ on Stack Overflow

how to create a dataframe class in pandas/python

1 Answer 1

Comments

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Related