我有一个嵌套在多个列表层中的列表列表.
可能的输入：
[[[[1,2,3],[a,b,c]]]]或[[[1,c]]]或[[[1,3] ]],[[a,c]]]
当我使用flat()时,它会压扁所有不是我想要的东西.
[1,3,A,B,C]
我需要的是
[[1,c]]
作为最终输出.
我的平面定义如下
def flat(S):
if S == []:
return S
if isinstance(S[0],list):
return flat(S[0]) + flat(S[1:])
return S[:1] + flat(S[1:])
解决方法
import collections
def is_listlike(x):
return isinstance(x,collections.Iterable) and not isinstance(x,basestring)
def flat(S):
result = []
for item in S:
if is_listlike(item) and len(item) > 0 and not is_listlike(item[0]):
result.append(item)
else:
result.extend(flat(item))
return result
tests = [ [[[[1,['a','b','c']]]],[[[1,'c']]],3]],[['a','c']]] ]
for S in tests:
print(flat(S))
产量
[[1,'c']] [[1,'c']]