从每个键具有多个唯一值的字符串列表创建一个 3 级字典

Question

我有一个文本字符串列表，我需要从中构建一棵树，据我了解，实现这一目标的正确数据结构是字典。 每个字符串的大小是固定的，所有元素的格式都是相同的，因此不需要额外的检查。 列表的每条记录都是格式为 DD/MM/YYYY 的日期，并且年/年应该在树的根部（键，这里没有重复），每年可能是多个月（内没有重复的月份同一年）作为值，每个月多天（同一个月内没有重复的天数）。

字符串列表示例：

数据 = ['04/02/2018', '05/02/2018', '06/02/2018', '01/03/2018', “2018 年 10 月 3 日”、“2017 年 8 月 9 日”、“2017 年 9 月 9 日”、“2017 年 11 月 10 日”、 '11/12/2017'、'14/06/2018'、'15/06/2018'、'24/07/2018'、'26/07/2018'、'30/08/2018'、'31 /08/2018', '01/09/2018']

除了解决方案，如果有的话，我还想解释一下以便理解。

这是我到目前为止写的，这显然是错误的，因为结果是一个只有最后 2 项的字典。

d = {}
for item in data:
    rec = item.split('/')
    d.update({rec[2]:{rec[1]:(rec[0])}})

该数据所需的输出如下所示：

{'2017': {'09': ['08', '09'], '10': ['11'], '12': ['11']},
 '2018': {'02': ['04', '05', '06'],
          '03': ['01', '10'],
          '06': ['14', '15'],
          '07': ['24', '26'],
          '08': ['30', '31'],
          '09': ['01']}}

Answer 1

你不需要一棵树。您可以使用 dicts 的 dict，将列表作为最里面的值。

您可以使用defaultdict 作为主要结构

result = defaultdict(defaultdict(list))

for date in data:
    day, month, year = date.split('/')
    result[year][month].append(day)

defaultdict 的作用是

• 对于内部字典：您可以直接append。如果没有列表，则会创建一个新的空列表。
• 对于外部字典：类似地，您可以参考键 month 假设已经有一个字典作为其值。如果没有，则创建一个新的。

Answer 2

有多种方法可以实现这一点。您可以使用来自collections 模块的defaultdict。但也可以使用普通的dict.setdefault 方法来完成。

setdefault(key[, default])

如果 key 在字典中，则返回其值。如果不是，则插入 key，其值为 default 并返回 default。 默认默认为无。

我们循环遍历数据，将其拆分为日、月和年字符串。然后我们在基础树中查找年份键，如果它不存在，我们为它创建一个新的空字典。然后我们在那个年份字典中查找月份键，如果它不存在则为其创建一个新列表。最后，我们将日期字符串附加到月份列表中。

from pprint import pprint

data = [
    '04/02/2018', '05/02/2018', '06/02/2018', '01/03/2018', '10/03/2018', '08/09/2017', '09/09/2017',
    '11/10/2017', '11/12/2017', '14/06/2018', '15/06/2018', '24/07/2018', '26/07/2018', '30/08/2018',
    '31/08/2018', '01/09/2018'
]

tree = {}

for s in data:
    day, mon, year = s.split('/')
    ydict = tree.setdefault(year, {})
    mlist = ydict.setdefault(mon, [])
    mlist.append(day)

pprint(tree)

输出

{'2017': {'09': ['08', '09'], '10': ['11'], '12': ['11']},
 '2018': {'02': ['04', '05', '06'],
          '03': ['01', '10'],
          '06': ['14', '15'],
          '07': ['24', '26'],
          '08': ['30', '31'],
          '09': ['01']}}

我们可以将主循环的 3 个步骤合并为一行，但是阅读起来有点困难：

for s in data:
    day, mon, year = s.split('/')
    tree.setdefault(year, {}).setdefault(mon, []).append(day)