通过属性访问嵌套dic的类通过print和__dict__显示不同的值



我正在处理一个项目,其中我需要具有一些属性(例如日期、名称(的对象,然后是一组值(通常是具有不同聚合的pd.DataFrames,但我将在此处使用整数作为示例(。与其进行多个嵌套dict并为键键入许多额外字符,如data["df"]["yearly"],不如使用data.df.yearly访问值。(有更好的结构吗?(

因此,我使用了一个将键设置为属性的类。但根据我访问数据的方式,我会得到不同的输出。

这是代码:

class Map(dict):
"""
https://stackoverflow.com/questions/2352181/how-to-use-a-dot-to-access-members-of-dictionary
Example:
m = Map({'first_name': 'Eduardo'}, last_name='Pool', age=24, sports=['Soccer'])
"""
def __init__(self, *args, **kwargs):
super(Map, self).__init__(*args, **kwargs)
for arg in args:
if isinstance(arg, dict):
for k, v in arg.items():
self[k] = v
if kwargs:
for k, v in kwargs.items():
self[k] = v
def __getattr__(self, attr):
super( Map, self ).__getattr__( attr )
return self.get(attr)
def __setattr__(self, key, value):
self.__setitem__(key, value)
def __setitem__( self, key, value ):
super( Map, self ).__setitem__( key, value )
if isinstance( value, dict ):
self.__dict__.update( {key: Map(value) } )         
else:
self.__dict__.update( {key: value} )
def __delattr__(self, item):
self.__delitem__(item)
def __delitem__(self, key):
super(Map, self).__delitem__(key)
del self.__dict__[key]   

现在,当我初始化一个实例并以以下方式为其分配一些数据时,我将有两个不同的输出,这取决于我是写data.df.yearly还是写data.df["yearly"]

有人看到错误了吗(或者有没有更好的数据结构的建议?(

创建Map((的实例

data = Map()
data.start_date = "2020-01-01"
data.finish_date = '2021-01-01'
data.name = "Type A"
data.df = {"yearly" : 0, "monthly" : 0, "weekly" : 0}

更改data.df 的值

for time in data.df.keys():
data.df[time] = 123

预期输出:

data
> {'start_date': '2020-01-01',
>  'finish_date': '2021-01-01',
>  'name': 'Type A',
>  'df': {'yearly': 123, 'monthly': 123, 'weekly': 123}}

输出(与打印(数据(相同(:

data
> {'start_date': '2020-01-01',
>  'finish_date': '2021-01-01',
>  'name': 'Type A',
>  'df': {'yearly': 0, 'monthly': 0, 'weekly': 0}}

通过dict的输出与所选输出匹配:

data.__dict__
> {'start_date': '2020-01-01',
>  'finish_date': '2021-01-01',
>  'name': 'Type A',
>  'df': {'yearly': 123, 'monthly': 123, 'weekly': 123}}

您的__setitem__方法使用原始值调用超类方法,但如果该值是字典,则在将其放入self.__dict__时会将其转换为Map。所以这两个位置包含不同的对象,没有任何东西可以保持它们的同步。

在将值存储在两个位置之前,您应该将其转换为Map

def __setitem__( self, key, value ):
if isinstance(value, dict):
value = Map(value)
super( Map, self ).__setitem__( key, value )
self.__dict__.update({key: value})

最新更新