将多维字典写入文件

Question

我有一个代码可以计算几个指标，这些指标存储为字典的 3 维字典。我想将这本词典打印到 csv 文件中 - 但还没有找到这样做的好方法。

计算完字典中的所有元素后，我想将其打印到文件中（其中不同的 periods 是文件的标题，keys 和指标 a, b, and c 应该是列 - 列键和列指标）。

有没有一种简单的方法可以将其打印到文件中？ （我的第一次尝试是 pandas，但这没有用）

谢谢

from collections import defaultdict
import pandas as pd
import os
import random


# 3 dimensional dictionary that stores integers 
output_dict = defaultdict(lambda: defaultdict(lambda: defaultdict(int)))
# Array of periods
periods = range(0, 2)
# relevant keys
keys = ["key1", "key2"]

# Iterate over all periods
for period in periods:
    # Iterate over all relevant keys
    for key in keys:

        # Store results for key for each time period for each category ("a", "b", or "c")
        output_dict[key][period]["a"] += random.randint(1, 1000)
        output_dict[key][period]["b"] += random.randint(1, 1000)
        output_dict[key][period]["c"] += random.randint(1, 1000)

# This is the tricky part!!!
# Store results 
pd.DataFrame(output_dict).to_csv("output_dict.csv", index=False)

# the dictionary may look as follows:
output_dict = {"key1": {0: {"a": 0.9, "b": 0.2, "c": 0.5}, 1:{"a": 0.91, "b": 0.3, "c": 0.4}},
               "key2": {0: {"a": 0.4, "b": 0.33, "c": 0.34}, 1: {"a": 0.21, "b": 0.73, "c": 0.54}}}

Answer 1

您应该为此使用 csv 模块，我认为不值得争论您的数据以使其与 pandas DataFrame 构造函数很好地配合使用。请注意，我将 csv 写入字符串 i/o 缓冲区而不是文件，因此我可以轻松打印结果，但您可以简单地省略这些内容并使用普通文件对象。

>>> periods = [0, 1]
>>> metrics = ['a', 'b', 'c']
>>> import csv
>>> import io

现在，只需仔细构建行：

>>> with io.StringIO() as f:
...     writer = csv.writer(f)
...     writer.writerow(['Key','Metric', 0, 1])
...     for key in output_dict:
...         for metric in metrics:
...             row = [key, metric]
...             for p in periods:
...                 row.append(output_dict[key][p][metric])
...             writer.writerow(row)
...     final = f.getvalue()
...
16
17
18
18
17
16
16
>>> print(final)
Key,Metric,0,1
key2,a,0.4,0.21
key2,b,0.33,0.73
key2,c,0.34,0.54
key1,a,0.9,0.91
key1,b,0.2,0.3
key1,c,0.5,0.4

请注意，键不会按任何特定顺序排列，因为字典是无序的。如果您提前知道它们，您可以通过迭代所有键来强加一个顺序，就像我对指标和周期所做的那样（您的问题暗示那些 是提前知道的）。这个解决方案可以很容易地扩展来处理丢失的键。

编辑： 您的最后一次编辑似乎暗示您将提前知道密钥，所以只需执行以下操作：

>>> periods = [0, 1]
>>> keys = ['key1', 'key2']
>>> metrics = ['a', 'b', 'c']
>>> with io.StringIO() as f:
...     writer = csv.writer(f)
...     writer.writerow(['Key','Metric', 0, 1])
...     for key in keys:
...         for metric in metrics:
...             row = [key, metric]
...             for p in periods:
...                 row.append(output_dict[key][p][metric])
...             writer.writerow(row)
...     final = f.getvalue()
...
16
17
16
16
17
18
18
>>> print(final)
Key,Metric,0,1
key1,a,0.9,0.91
key1,b,0.2,0.3
key1,c,0.5,0.4
key2,a,0.4,0.21
key2,b,0.33,0.73
key2,c,0.34,0.54