自定义pandas.DataFrame类

时间:2018-12-31 19:02:02

标签: python pandas oop dataframe inheritance

我已经创建了一个基于pandas.DataFrame对象的类,该对象使用csv文件进行了初始化,如下所示:

import pandas as pd

class CustomDataFrame(pd.DataFrame):

    def  __init__(self, input_file):
        df = pd.read_csv(input_file)
        super().__init__(df)
    #...

这样,我有一个CustomDataFrame类型,它具有其他可对其自身进行操作的特定方法。此设置的问题在于,当我对对象进行切片时,它将返回pandas.DataFrame对象,而不是保持相同的类型。换句话说:

> blip = mypackage.CustomDataFrame('test.csv')

> type(blip)
mypackage.CustomDataFrame

> type(blip[1:3])
pandas.core.frame.DataFrame

是否有一种简单的方法来更正我的自定义类,以便它可以以pandas.DataFrame的所有方式对其自身进行操作,同时每次都返回此自定义类,而不仅仅是返回内置类DataFrame

0 个答案:

没有答案