如何使用 python 使我的代码更简单

Question

这是我的代码：

def set_floor_point(self,floor_point=None):
    if self.data.get('stage'):
        self.data['stage'] = {}
    stage_number = self.get_stage_number()
    floor_number = self.get_floor_number()

    if self.data['stage'].get(stage_number):
        self.data['stage'][stage_number] = {}

    if self.data['stage'][stage_number].get('floor_point'):
        self.data['stage'][stage_number]['floor_point'] = {}

    if self.data['stage'][stage_number]['floor_point'].get(floor_number):
        self.data['stage'][stage_number]['floor_point'][floor_number] = {}


    self.data['stage'][stage_number]['floor_point'][floor_number] = floor_point

我第一次创建的字典是这样的：

stage = 
{
    0:{
        'floor':{
            0:{
                'floor_point':0,
                'gift':{}
            }
        }
    }
}

但是我觉得我的代码不是很好，太繁琐了，

所以有人知道更简单的方法吗，

谢谢

Answer 1

data = collections.defaultdict(lambda: collections.defaultdict(
  lambda: collections.defaultdict(dict)))
data['stage'][3]['floor_point'][2] = 5
print data

Answer 2

我不确定你想要实现什么。您的代码中反复出现的主题是：

if some_dict.get(key):
    some_dict[key] = {}

这意味着：如果some_dict 有一个键key 并且some_dict[key] 是一个真实值，那么将some_dict[key] 替换为{}。如果some_dict 没有键key 或some_dict[key] 是一个假值（None、0、False、[] 等），则什么也不做。

如果这是你想要的，你可以这样澄清你的：

def replace_value_by_empty_dict(d, key):
    if d.get(key):
        d[key] = {}
...
replace_value_by_empty_dict(self.data, 'stage')
etc.

但如果这不是您想要的（如果ifs 之一为真，则代码将中断），您可能需要用英文单词或伪代码来表述问题以阐明问题的结构。

看看collections.defaultdict。