在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]
这样,嵌套的列表就被展平为一维列表了。需要注意的是,展平函数会递归地处理嵌套的列表,直到所有的嵌套层级都被展平为一维列表。