温馨提示×

python中flatten函数的用法是什么

小亿
186
2024-03-04 18:59:58
栏目: 编程语言

在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]

这样,嵌套的列表就被展平为一维列表了。需要注意的是,展平函数会递归地处理嵌套的列表,直到所有的嵌套层级都被展平为一维列表。

0