我有如下的 python 代碼：import numpy as npimport multiprocessing as mpdef 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 resultswith 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()之后，當(dāng)我打印res輸出時(shí)，如下所示：[[{'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}]您知道如何更改代碼以獲得所需的結(jié)果嗎res？
2 回答
泛舟湖上清波郎朗
TA貢獻(xiàn)1818條經(jīng)驗(yàn) 獲得超3個(gè)贊
of庫允許您將chain.from_iterable子itertools列表鏈接到一個(gè)列表。
import itertools
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}]]
print(list(itertools.chain.from_iterable(res)))
楊魅力
TA貢獻(xiàn)1811條經(jīng)驗(yàn) 獲得超6個(gè)贊
正如在您的示例中，您將字典存儲(chǔ)在列表列表中，列表理解也有效：
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}]
添加回答
舉報(bào)
0/150
提交
取消