问题描述
如何使单个实例(如 int 或 str)可迭代。又名,我如何展平可能包含嵌套列表的列表。 这个很重要。考虑到列表(或函数返回)可以包含一个可迭代对象,如列表或生成器,但它也可以包含单个值。我必须处理它们。 在下面的示例中,我有一个包含单数元素和元素列表的列表。我只想将它们展平为一个简单的列表。 我该怎么做?
org_l = [[1,2,3,4],[5],6,7,[8,9,0]]
new_l = [e1 for e in l for e1 in e]
Traceback (most recent call last):
File "C:\Users\krisv\PycharmProjects\pythonProject\main.py",line 101,in <module>
l = [e1 for e in l for e1 in e]
File "C:\Users\krisv\PycharmProjects\pythonProject\main.py",in <listcomp>
l = [e1 for e in l for e1 in e]
TypeError: 'int' object is not iterable
需要的是 new_l 等于: [1,4,5,8,0]
解决方法
我会回答我自己的问题:
def unroll(thing):
if isinstance(thing,typing.Iterable) and not isinstance(thing,str):
for e in thing:
yield from unroll(e)
else:
yield thing
l = [[1,2,[3,'c'],4],[5],6,7,7.5,[8,9,9.5,0]]
flat_list = [e for e in unroll(l)]
print(flat_list)
>>>
[1,3,'c',4,5,8,0]