![](/img/trans.png)
[英]Flatten a list of lists in Python Flatten() does not work Possible generator issue
[英]python: flatten to a list of lists but no more
我有一個嵌套在多個列表層中的列表列表。
可能的輸入:
[[[[1,2,3] , [a,b,c]]]]
或[[[1,2,3] , [a,b,c]]]
或[[[1,2,3]] , [[a,b,c]]]
當我使用flat()
它會壓扁所有不是我想要的東西。
[1,2,3,a,b,c]
我需要的是
[[1,2,3] , [a,b,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,2,3] , ['a','b','c']]]],
[[[1,2,3] , ['a','b','c']]],
[[[1,2,3]] , [['a','b','c']]] ]
for S in tests:
print(flat(S))
產量
[[1, 2, 3], ['a', 'b', 'c']]
[[1, 2, 3], ['a', 'b', 'c']]
[[1, 2, 3], ['a', 'b', 'c']]
更換:
if S == []:
return S
有:
if (not any([isinstance(x,list) for x in S])) :
return [] if S==[] else [S]
似乎可以做到這一點。
要么:
if S == []:
return S
if (not any([isinstance(x,list) for x in S])) :
return [S]
我看到兩個要求 - 檢測S
何時不應該被展平,然后返回一個在與其余部分連接時不會變平的值(即使用append
而不是extend
)。 我的猜測是,非列表元素列表不應該被展平。
我和unutbu
一樣吠叫同一棵樹,但是更加混亂。:)
flat()可能會成為
def flat(mylist):
return[val for sublist in mylist for val in sublist]
然后你可以像這樣在for循環中調用它
while type(mylist[0][0]) is list:
mylist = flat(my list)
無論嵌套列表的數量如何,它都會將其減少到您想要的輸出
[[1, 2, 3], ['a', 'b', 'c']]
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.