python - 使用函数在类中设置列表名称

Question

我有一个从 CSV 文件导入数据的类，以及一个获取文件名和输出列表名称的函数。我想使用 setattr() 函数将 self.data_name 的名称设置为 self.info 。我怎样才能做到这一点？

import csv

class import_data:

    def import_csv(self, filename_csv, data_name):


            setattr(self,data_name,0)

            datafile = open(filename_csv, 'r')
            datareader = csv.reader(datafile)
            self.data_name = []

            for row in datareader:
                self.data_name.append(row)
            print("finished importing data")

b = import_data()
b.import_csv('info.csv', 'info')
print(b.info)

这不起作用，因为 b.data_name 不是 b.info。这将打印 0 而不是导入的 CSV 文件。

Answer 1

试试这个:

class import_data:
    def import_csv(self, filename_csv, data_name):
        with open(filename_csv, 'r') as f:
            setattr(self, data_name, list(csv.reader(f)))
        print("finished importing data")