Python3教程：字典转为成员变量的方法

Posted 2022-06-25 Python热爱者

当我们在Python中写一个class时，如果有一部分的成员变量需要用一个字典来命名和赋值，此时应该如何操作呢？这个场景最常见于从一个文件（比如json、npz之类的文件）中读取字典变量到内存当中，再赋值给一个类的成员变量，或者已经生成的实例变量。

使用​​__dict__​​定义成员变量

在python中直接支持了​​__dict__.update()​​这样的方法来操作，避免了对locals()、vars()和eval()函数的使用，我们可以直接看这样的一个案例：

In [1]: dict_a = a:1, b:2

In [2]: dict_b = c:3

In [3]: class D(object):
   ...:     def __init__(self):
   ...:         self.d = 4
   ...:         self.__dict__.update(dict_a)
   ...:         self.__dict__.update(dict_b)
   ...: 

In [4]: new_D = D()

In [5]: new_D.__dict__
Out[5]: d: 4, a: 1, b: 2, c: 3

In [6]: new_D.a
Out[6]: 1

In [7]: new_D.c
Out[7]: 3

在这个案例中，我们在类的外部定义了两个字典​​dict_a​​​和​​dict_b​​​，字典的key值都是字符串的格式。而我们知道字符串格式在python中如果不使用eval，是不能直接作为变量名来使用的。而通过​​__dict__.update()​​的方法将字典导入后，会自动识别其中所有的key和value值，转而赋值给当前类作为成员变量。但是这个方法有一个缺点是，只能通过单层的字典来赋值，如果遇到有层级结构的字典，是不会自动分辨层级结构进行赋值的，比如下方的代码：

In [15]: dict_a = f:h:8,g:7

In [16]: new_D = D()

In [17]: new_D.__dict__
Out[17]: d: 4, f: h: 8, g: 7, c: 3

嵌套字典转成员变量

根据上一个章节中最后提到的这个特殊场景，我们需要对字典内的元素进行递归，如果遇到嵌套的字典元素，则递归的将该元素添加到下一层级的成员变量中，具体代码如下所示：

学习中遇到问题没人解答？小编创建了一个Python学习交流QQ群：857662006
寻找有志同道合的小伙伴，互帮互助,群里还有不错的视频学习教程和PDF电子书！


dict_a = f:h:8,g:7
dict_b = c:3

class D:
    def __init__(self, *args):
        for arg in args:
            for k, v in arg.items():
                if isinstance(v, dict):
                    self.__dict__[k] = D(v)
                else:
                    self.__dict__[k] = v

new_D = D(dict_a, dict_b)
print (new_D.__dict__)
print (new_D.f.h)

最终的输出结果如下所示：

f: <__main__.D object at 0x7fd2f32a4340>
, g: 7, c: 3
8

可以看到，我们最终通过​​new_D.f.h​​​的方法，成功读取了原有的嵌套字典中的value。虽然这种写法看起来不是很优雅，但是似乎也没有更好的解决方案。并且，通过这个小问题的实践，发现了另外一个略有意思的问题：对python中的字典类型进行更新时，如果key的字符串中带有点号，比如​​parDict[group1.b] = 3​​​，只能通过这样的字符串的形式进行更新，如果使用​​parDict.update(group1.b=4)​​则会发生报错，这是因为点号在python中不是一个标识符，不能用于命名，原文内容如下：

The valid characters for identifiers are the same as in Python 2.x:
the uppercase and lowercase letters A through Z, the underscore _ and,
except for the first character, the digits 0 through 9.

总结

本文所解决的问题场景是这样的：如果给定一个字典，比如一般从json文件或者npz文件中加载出来的数据都是字典的数据结构，如果我们希望把这个字典赋值给一个类，且使得字典的key和value分别作为类的成员变量名和成员变量值，那么该如何实现呢？

对于一个展平的字典而言（没有嵌套字典），我们直接使用update就可以将字典中的所有key和value转变为当前类的成员变量。比较麻烦的是包含有嵌套字典的层级结构字典，此时我们只能使用循环，并且递归的对类的成员变量进行赋值。