python - 读取文件内容的类

Question

我想编写读取 *.csv 文件并使用 pandas 库解析它的类。我想知道应该在哪里初始化 df.

#!/usr/bin/env python
import pandas as pd
import os

class ParseDataBase(object):
        def __init__(self, name_file):
                self.name_file = name_file

        def read_file(self):
                """Read the file concent"""

                try:
                        self.df = pd.read_csv(self.name_file)
                except IndexError:
                        print ("Error: Wrong file name")
                        sys.exit(2)
                return self.df

        def dispaly_file(self):
                print self.df

def main():
        x = ParseDataBase('something.csv')
        x.dispaly_file()

if __name__ == '__main__':
        main()

上述代码返回以下错误:“ParseDataBase”对象没有属性“df”。

我不想在创建对象时传递很多变量。

我是面向对象编程的新手，因此非常感谢任何评论和提示!

Answer 1

除非您运行 read_file()，否则您不会分配 self.df，但事实并非如此。

def main():
    x = ParseDataBase('something.csv')
    x.read_file()     
    x.dispaly_file()