我已经创建了一个基于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
?