正如評論中指出的那樣，對于這個相關問題的大多數（如果不是全部）答案都失敗了，例如：ls = [1,2,[3,4]]此外，該列表可以嵌套更深。如何部分展平到用戶給定的級別（默認為無窮大）ls2 = [1,[2,3],[4,[5,6]]]所需的輸出ls2：展平到級別 1： [1,2,3,4,[5,6]]展平到 2 級（或更高） [1,2,3,4,5,6]
2 回答
三國紛爭
TA貢獻1804條經驗 獲得超7個贊
你可以遞歸地做到這一點：
def flatten(l, level=None):
if level == 0:
return l
flattened = []
for item in l:
if isinstance(item, list):
flattened.extend(flatten(item, level-1 if level is not None else None))
else:
flattened.append(item)
return flattened
ls2 = [1,[2,3],[4,[5,6]]]
print(flatten(ls2, level=1))
# [1, 2, 3, 4, [5, 6]]
print(flatten(ls2, level=2))
# [1, 2, 3, 4, 5, 6]
print(flatten(ls2))
# [1, 2, 3, 4, 5, 6]
largeQ
TA貢獻2039條經驗 獲得超8個贊
一種方法
ls2 = [1,[2,3],[4,[5,6]]]
def make_list_of_list(a):
return [[i]if not isinstance(i, list) else i for i in a]
def flatten(l):
return [item for sublist in make_list_of_list(l) for item in sublist]
flatten(ls2) will result in [1, 2, 3, 4, [5, 6]]
flatten(flatten(ls2)) will result in [1, 2, 3, 4, 5, 6]
添加回答
舉報
0/150
提交
取消