在Python中,flatten函數通常用來將嵌套的列表(或其他可迭代對象)展平為一維列表。這在處理嵌套結構數據時非常有用,可以簡化數據處理和操作。
以下是一個示例展示如何編寫一個flatten函數:
def flatten(lst):
flat_list = []
for item in lst:
if isinstance(item, list):
flat_list.extend(flatten(item))
else:
flat_list.append(item)
return flat_list
使用這個flatten函數,可以將嵌套的列表展平為一維列表。例如:
nested_list = [1, [2, 3], [4, [5, 6]]]
flattened_list = flatten(nested_list)
print(flattened_list)
# 輸出:[1, 2, 3, 4, 5, 6]
這樣,嵌套的列表就被展平為一維列表了。需要注意的是,展平函數會遞歸地處理嵌套的列表,直到所有的嵌套層級都被展平為一維列表。