python - 希望实例 df = MyClass() 成为显示键入 df 的熊猫数据框

Question

这是一个初学者的问题。我想写一个继承自 pandas DataFrame 的类

import pandas as pd

class MyClass(pd.DataFrame):
    # my code

一个实例

df = MyClass()

应该是由返回的数据框pd.read_csv('myfile.csv') ，我希望它在简单输入时显示

df

在 iPython 下，当然，pd.DataFrame 的所有方法都可以工作（除了__init__我需要重写的方法）。我不知道如何获得所需的行为。我所有的尝试都失败了。

标签： pythonpandasinheritance

Answer 1

An__init__和super函数是您正在寻找的：

class MyClass(pd.DataFrame):
    def __init__(self,file_name = 'myfile.csv'):
        super().__init__(pd.read_csv(file_name))
        #python 3

什么是__init__和super？点击了解（python 文档）。