博客

python中flatten()函数用法详解

  • Post category:Python

当然，我很乐意为您提供“Python中flatten()函数用法详解”的完整攻略。以下是详细的步骤和示例：

flatten()函数

在Python中，flatten函数用于将嵌套的列表或元组展开为一维列表。flatten()函数可以递归地展开嵌套的列表或元组，直到嵌套的列表或元组都被展开为一维列表。

语法

flatten(lst, ltypes=(list, tuple)):

参数

  • lst：需要展开的列表或元组。
  • ltypes：需要展开的列表或元组的类型，默认为(list, tuple)。

返回值

展开后的一维列表。

示例1：展开嵌套列表

def flatten(lst, ltypes=(list, tuple)):
    if not isinstance(lst, ltypes):
        return [lst]
    result = []
    for item in lst:
        result.extend(flatten(item, ltypes))
    return result

lst = [1, 2, [3, 4, [5, 6], 7], 8]
print(flatten(lst))

在这个示例中，我们定义了一个名为flatten()函数，用于展开嵌套的列表。我们使用isinstance()函数判断是否为列表或元组类型，如果不是，则将lst作为一个元素添加到结果列表中。如果lst是列表或元组类型，则遍历lst中的每个元素，并递归调用flatten()函数展开嵌套的列表。我们使用extend()函数将展开后的一维列表添加到结果列表中，并返回结果列表。

示例2：展开嵌套的元组

def flatten(lst, ltypes=(list, tuple)):
    if not isinstance(lst, ltypes):
        return [lst]
    result = []
    for item in lst:
        result.extend(flatten(item, ltypes))
    return result

lst = (1, 2, (3, 4, (5, 6), 7), 8)
print(flatten(lst))

在这个示例中，我们使用flatten()函数展开了嵌套的元组。我们使用与示例1相同的方法展开嵌套的元组，并使用print()函数输出展开后的一维列表。

以上是“Python中flatten()函数用法详解”的完整攻略，我们使用两个示例演示了如何使用flatten()函数展开嵌套的列表或元组。flatten()函数是一个非常有用的函数，可以简化嵌套列表或元组的处理。

Tags: