如何展平包含具有不同数量词典的列表的列表？

Question

我有如下的 python 代码：

import numpy as np

import multiprocessing as mp

def func(first, sec):

    results = []

    for x in range(first):

        result_dict = {"a": x, "b": x**sec, "c": x/sec}

        results.append(result_dict.copy())

    return results

with mp.Pool(mp.cpu_count()) as pool:

    res = pool.starmap(func, [(a, 8) for a in range(1, 4)])

然后我res用这段代码展平了：

res = np.asarray(res)

res = res.reshape(-1)

res = np.array(res).tolist()

之后，当我打印res输出时，如下所示：

[[{'a': 0, 'b': 0, 'c': 0.0}],

 [{'a': 0, 'b': 0, 'c': 0.0}, {'a': 1, 'b': 1, 'c': 0.125}],

 [{'a': 0, 'b': 0, 'c': 0.0},

  {'a': 1, 'b': 1, 'c': 0.125},

  {'a': 2, 'b': 256, 'c': 0.25}]]

但我希望输出是这样的：

[{'a': 0, 'b': 0, 'c': 0.0},

 {'a': 0, 'b': 0, 'c': 0.0},

 {'a': 1, 'b': 1, 'c': 0.125},

 {'a': 0, 'b': 0, 'c': 0.0},

 {'a': 1, 'b': 1, 'c': 0.125},

 {'a': 2, 'b': 256, 'c': 0.25}]

您知道如何更改代码以获得所需的结果吗res？

Answer 1

of库允许您将chain.from_iterable子itertools列表链接到一个列表。import itertoolsres = [[{'a': 0, 'b': 0, 'c': 0.0}], [{'a': 0, 'b': 0, 'c': 0.0}, {'a': 1, 'b': 1, 'c': 0.125}], [{'a': 0, 'b': 0, 'c': 0.0},  {'a': 1, 'b': 1, 'c': 0.125},  {'a': 2, 'b': 256, 'c': 0.25}]]print(list(itertools.chain.from_iterable(res)))

Answer 2

正如在您的示例中，您将字典存储在列表列表中，列表理解也有效：res = [[{'a': 0, 'b': 0, 'c': 0.0}], [{'a': 0, 'b': 0, 'c': 0.0}, {'a': 1, 'b': 1, 'c': 0.125}], [{'a': 0, 'b': 0, 'c': 0.0},  {'a': 1, 'b': 1, 'c': 0.125},  {'a': 2, 'b': 256, 'c': 0.25}]]flat_res = [item for items in res for item in items]输出print(flat_res)# [{'a': 0, 'b': 0, 'c': 0.0}, {'a': 0, 'b': 0, 'c': 0.0}, {'a': 1, 'b': 1, 'c': 0.125}, {'a': 0, 'b': 0, 'c': 0.0}, {'a': 1, 'b': 1, 'c': 0.125}, {'a': 2, 'b': 256, 'c': 0.25}]