温馨提示×

python中flatten函数的用法是什么

小亿
202
2024-03-04 18:59:58
栏目: 编程语言
Python开发者专用服务器限时活动,0元免费领,库存有限,领完即止! 点击查看>>

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

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

亿速云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>

推荐阅读:pytorch中flatten函数的用法是什么

0