defaultdict(list) 将所有值连接到一个列表中

Question

我想做什么：

编写一个方法来对一个字符串数组进行排序，这样所有的 anagrms 都是 彼此相邻。

我有以下代码：

from collections import defaultdict
res = defaultdict(list)
L = ['foo', 'poo', 'k', 'fo', 'ofo', 'oof']

for w in L:
    res["".join(sorted(w))].append(w)

但现在我想将res 中的所有值合并到一个列表中。

我试过了：

output =[]
for items in res.values():
    output.append(i for i in items)

但这给了我：

>>> output
[<generator object <genexpr> at 0x102a4d1e0>, <generator object <genexpr> at 0x102a95870>, <generator object <genexpr> at 0x102a958c0>, <generator object <genexpr> at 0x102a95910>]

如何正确显示一个列表中的项目？

期望：

['foo','ofo', 'oof','poo', 'k', 'fo',]

（所有字谜都在一起，顺序无关紧要，只要它们在列表中相邻即可。）

Answer 1

当你这样做时 -

output.append(i for i in items)

您实际上是将生成器表达式 - i for i in items - 附加到 output 列表中，您也可以从 output 列表的 repr 结果中看到。这不会自动评估生成器表达式并添加结果。你应该做的是使用output.extend()。示例 -

>>> output = []
>>> for items in res.values():
...     output.extend(items)
...
>>> output
['fo', 'k', 'foo', 'ofo', 'oof', 'poo']

这将评估表达式并将可迭代（生成器表达式）中的每个元素作为单独的元素附加到列表中。

---

但是如果你想从res字典的所有列表值转换单个列表，一个更简单的方法是使用itertools.chain.from_iterable。示例 -

>>> from itertools import chain
>>> output = list(chain.from_iterable(res.values()))
>>> output
['fo', 'k', 'foo', 'ofo', 'oof', 'poo']

Answer 2

已编辑

我相信替换

output.append(i for i in items)

与

output.extend(items)

会做你期望的。