Question

我有一个具有以下结构的字典：

{ "123" : {"red" : ['some text', datetime.datetime(2011, 8, 23, 3, 19, 38), status]},
  "456" : {"red" : ['some other text', datetime.datetime(2013, 8, 23, 3, 19, 38), status],
           "blue" : ['some more text', datetime.datetime(2010, 8, 23, 3, 19, 38), status]},
  "789" : {"blue" : ['random text', datetime.datetime(2012, 8, 23, 3, 19, 38), status],
           "yellow" : ['text', datetime.datetime(2009, 8, 23, 3, 19, 38), status]}}

现在我有一些逻辑来更新这本词典。它首先检查该字典中是否已存在条目，如果存在，是否存在子条目并比较时间和更新。如果其中一个不存在，则会创建一个新条目：

if example_id in my_directory:
    if color in my_directory[example_id]:
        if time > my_directory[example_id][color][1]:
            my_directory[example_id][color] = [text, time, status]
    else:
        my_directory[example_id] = {color : [text, time, status]}
else:
    my_directory[example_id] = {color : [text, time, status]}

显然，time，color和status作为已存在的变量传递。重写此IF语句以不复制第二个和第三个目录更新命令的正确方法是什么？谢谢！

Answer 1

正如其他人所说，使用defaultdict：

my_dictionary = collections.defaultdict(
                lambda: collections.defaultdict(
                lambda: (None, datetime.datetime.min, None)))

# populate my_dictionary

_, old_time, _ = my_dictionary[example_id][color]
if time > old_time:
    # NB: tuples make more sense here than lists
    my_directory[example_id][color] = (text, time, status)

这会暂时将(None, datetime.datetime.min, None)元组添加到您的字典中，然后将其替换为实际值。

Answer 2

您可以使用and操作并删除额外的else：

if example_id in my_directory and color in my_directory[example_id] and time > my_directory[example_id][color][1]:
            my_directory[example_id][color] = [text, time, status]
else:
            my_directory[example_id] = {color : [text, time, status]}

在Python中重写这个嵌套IF语句的正确方法是什么

2 个答案: