python - 将 Python 嵌套字典转换为 Numpy 数组

Question

我有一个格式为的字典:

{"k": {"k1": [[v1]],"k2": [[v2]],"k3": [[v3]]}

如何将其转换为以下格式的两个数组:

[kk1,kk2,kk3]
[v1,v2,v3]

我有一个 Python 函数，但是我无法获得所需格式的结果。

def dict_to_list_of_keys(d, l, loc):
    for k in iter(d):
        loc.append(k)
        l.append(loc * 1)
        if isinstance(d[k], dict):
            dict_2_list_of_keys(d[k], l, loc)
        loc.pop()
    return l

Answer 1

将 for 循环与 list.append 结合使用。将列表转换为数组很简单，即 np.array(L)。

d = {"k": {"k1": [['v1']],"k2": [['v2']],"k3": [['v3']]}}

keys, values = [], []

for k1, v1 in d.items():
    for k2, v2 in v1.items():
        keys.append(k1+k2)
        values.append(v2[0][0])

print(keys, values, sep='\n')

['kk1', 'kk2', 'kk3']
['v1', 'v2', 'v3']