如何在Python中将列表转换为嵌套字典

Question

需要将 x:

X = [['A', 'B', 'C'], ['A', 'B', 'D']]

转换为 Y:

Y = {'A': {'B': {'C','D'}}}

更具体地说，我需要从绝对路径列表创建文件夹和文件树，如下所示：

paths = ['xyz/123/file.txt', 'abc/456/otherfile.txt']

其中，每个路径都是 split("/")，如伪示例中的 ['A', 'B', 'C'] 所示。

由于这代表文件和文件夹，显然，在同一级别（数组的索引）上，同名字符串不能重复。

Answer 1

X = [['A', 'B', 'C'], ['A', 'B', 'D'],['W','X'],['W','Y','Z']]
d = {}

for path in X:
    current_level = d
    for part in path:
        if part not in current_level:
            current_level[part] = {}
        current_level = current_level[part]

这使得 d 包含 {'A': {'B': {'C': {}, 'D': {}}}, 'W': {'Y': {'Z': {}}，“X”：{}}}。任何包含空字典的项目要么是文件，要么是空目录。

Answer 2

假设 {'C', 'D'} 表示 set(['C', 'D']) 并且您的 Python 版本支持字典理解 和 设置理解，这是一个丑陋但有效的解决方案：

>>> tr = [[1, 2, 3], [1, 2, 4], [5, 6, 7]]
>>> {a[0]: {b[1]: {c[2] for c in [y for y in tr if y[1] == b[1]]} for b in [x for x in tr if x[0] == a[0]]} for a in tr}
{1: {2: set([3, 4])}, 5: {6: set([7])}}

至于您的示例：

>>> X = [['A', 'B', 'C'], ['A', 'B', 'D']]
>>> {a[0]: {b[1]: {c[2] for c in [y for y in X if y[1] == b[1]]} for b in [x for x in X if x[0] == a[0]]} for a in X}
{'A': {'B': set(['C', 'D'])}}

但请不要在现实世界的应用程序中使用它:)

更新： 这里是一个可以任意使用的深度：

>>> def todict(lst, d=0):
...     print lst, d
...     if d > len(lst):
...         return {}
...     return {a[d]: todict([x for x in X if x[d] == a[d]], d+1) for a in lst}
...
>>> todict(X)
{'A': {'B': {'C': {}, 'D': {}}}}

Answer 3

您的问题陈述存在逻辑不一致。如果你真的想要
['xyz/123/file.txt', 'abc/456/otherfile.txt']

更改为
{'xyz': {'123': 'file.txt}, 'abc': {'456': 'otherfile.txt'}}

然后你必须回答路径 'abc.txt' 是如何实现的。没有前导文件夹的“txt”将被插入到此数据结构中。顶级字典键会是空字符串 '' 吗？

Answer 4

这应该非常接近您所需要的：

def path_to_dict(path):
    parts = path.split('/')

    def pack(parts):
        if len(parts) == 1:
            return parts
        elif len(parts):
            return {parts[0]: pack(parts[1:])}
        return parts

    return pack(parts)

if __name__ == '__main__':
    paths = ['xyz/123/file.txt', 'abc/456/otherfile.txt']
    for path in paths:
        print '%s -> %s' % (path, path_to_dict(path))

结果：

xyz/123/file.txt -> {'xyz': {'123': ['file.txt']}}
abc/456/otherfile.txt -> {'abc': {'456': ['otherfile.txt']}}

Answer 5

我在 Twitter 上被问到这个问题，并使用函数式编程想出了这个巧妙的解决方案，我想我也可以在这里分享。

from functools import reduce
X = [['A', 'B', 'C'], ['A', 'B', 'D']]
Y = [reduce(lambda x, y: {y:x}, Y[::-1]) for Y in X]

返回：

[{'A': {'B': 'C'}}, {'A': {'B': 'D'}}]

根据需要。

对于更简单的问题，您有一个列表想要表示为带有嵌套键的字典，这就足够了：

from functools import reduce
X = ['A', 'B', 'C']
reduce(lambda x, y: {y:x}, X[::-1])

它返回：

{'A': {'B': 'C'}}

Answer 6

首先将键与值分开

x = [['A', 'B', 'C'], ['A', 'B', 'D']]
keys = [tuple(asd[:-1]) for asd in x]
values = [asd[-1] for asd in x]

现在使用它们来填充 NestedDict

from ndicts.ndicts import NestedDict

nd = NestedDict()
for key, value in zip(keys, values):
    nd[key] = value

>>> nd
NestedDict({'A': {'B': 'D'}})
>>> nd.to_dict()
{'A': {'B': 'D'}}

要安装 ndicts pip 安装 ndicts