需求是:
一个多层嵌套的列表,每个嵌套列表有2个列表元素(注意这里实现的是两个元素),将多个嵌套展开成一层列表嵌套。目的是,保留最后的一层列表结构。
示例输入和输出:
输入:[[[[‘ABC’, ‘efg’], [‘T’, ‘C’]], [‘d’]], [‘c’]]
输出:[[‘ABC’, ‘efg’], [‘T’, ‘C’], [‘d’], [‘c’]]
注意: 这里输入的列表中有3层嵌套,希望输出的列表只有1层嵌套,且输入的列表每层嵌套中,只有两个列表元素。
a = [[[['ABC', 'efg'], ['T', 'C']], ['d']], ['c']] # 嵌套在前
def simple_lst(lst, new_lst):
if isinstance(lst[1], list):
new_lst.append(lst[1])
else:
new_lst.append(lst)
if isinstance(lst[0], list):
simple_lst(lst[0], new_lst)
return new_lst[::-1]
alst = simple_lst(a, [])
print(alst)
思考:
1) 如果是嵌套在后,如何修改?
b = [['ABC', 'efg'], [['T', 'C'], [['d'], ['c']]]] # 嵌套在后
2) 如果嵌套的元素列表个数>2,如何实现?
c = [[[['ABC', 'efg'], ['T', 'C'], ['x', 'y']], ['d'], ['z']], ['c'], ['s']]
- 如果是需要把多层嵌套的列表展成无嵌套的列表,例如:
输入:[[[1,2], [3]], [4]]
输出:[1,2,3,4]
def dim_lst(lst, new_lst):
"""
lst = [[1,2], [3], [5]]
return [1,2,3,5]
lst = [[[1,2], [3]], [4]]
return [1,2,3,4]
"""
for i in lst:
if not isinstance(i, list):
new_lst.append(i)
else:
dim_lst(i, new_lst)
return new_lst