将数据存储在类中，并动态访问数据作为类属性

Question

我正在尝试编写一个类，该类将数据框架ID作为字符串和值作为数据框的值，并创建访问数据的类属性。

我能够编写一个类似类的小例子，该类别需要以静态方式创建的方法并将对象返回为类方法，但我想循环浏览数据，以 df的键键入键 s，并允许使用属性访问每个 df 。

最低工作示例

from dataclasses import dataclass
import pandas as pd

# re-writing as dataclass
@dataclass
class Dataset:

    # data container dictionary as class attribute
    dict = {'df1_id':pd.DataFrame({'col1':[1,1]}), 
            'df2_id':pd.DataFrame({'col2':[2,2]}), 
            'df3_id':pd.DataFrame({'col3':[3,3]})}
    
    
    def df1_id(self) -> pd.DataFrame:# class method to create as class attribute
        return dict['df1_id']

    def df2_id(self) -> pd.DataFrame:# same class method above
        return dict['df2_id']

    def df3_id(self) -> pd.DataFrame:# same class method above
        return dict['df3_id']

    def dataframes_as_class_attributes(self):
        # store the dfs to access as class attributes
        # replacing 3 methods above
        return 

结果

datasets = Dataset()

print(datasets.df1_id())

预期结果

datasets = Dataset()

print(datasets.df1_id) # class attribute created by looping through the dict object

编辑：

• 与此相似：如何将CSV文件的内容读取到一个将每个CSV行作为类实例上课

Answer 1

摄入 df s的键，并允许使用属性访问每个 df 。

似乎班级的唯一目的是具有属性访问语法。在这种情况下，只需创建a >对象。

from types import SimpleNamespace

class Dataset(SimpleNamespace):
    pass 
    # extend it possibly 

data = {
    'df1_id':pd.DataFrame({'col1':[1,1]}), 
    'df2_id':pd.DataFrame({'col2':[2,2]}), 
    'df3_id':pd.DataFrame({'col3':[3,3]})
}

datasets = Dataset(**data)

输出：

>>> datasets.df1_id 

   col1
0     1
1     1

>>> datasets.df2_id 

   col2
0     2
1     2

>>> datasets.df3_id 

   col3
0     3
1     3

Answer 2

您可以使用 setAttr 如下所示：

from dataclasses import dataclass
import pandas as pd

@dataclass
class Dataset:

    dict_ = {'df1_id':pd.DataFrame({'col1':[1,1]}), 
            'df2_id':pd.DataFrame({'col2':[2,2]}), 
            'df3_id':pd.DataFrame({'col3':[3,3]})}
            
    def __post_init__(self):
        for key, val in self.dict_.items():
            setattr(self, key, val)

为了避免与Python关键字发生冲突，将单个尾随下划线放置在变量名称之后。 （ pep 8 ）