Python | 扁平化二维列表的方法

在Python中，我们可以使用不同的方法来扁平化一个二维列表。扁平化二维列表是指将一个包含多个列表的列表转换为一个单一的列表，其中包含所有原始列表中的元素。本文将介绍几种用于扁平化二维列表的不同方法。

Method 1：使用迭代方法

这种方法使用迭代来遍历列表中的每个元素，并将它们添加到一个新的列表中。以下是使用迭代方法的示例代码：

def flatten_2d_list(lst):
    result = []
    for elem in lst:
        if type(elem) == list:
            result += flatten_2d_list(elem)
        else:
            result.append(elem)
    return result

Method 2：使用列表解析

列表解析是一种简洁的方法，它允许我们在单个语句中生成新列表。以下是使用列表解析的示例代码：

def flatten_2d_list(lst):
    return [elem for sublist in lst for elem in sublist]

Method 3：使用嵌套列表解析

我们也可以使用嵌套列表解析来扁平化二维列表。以下是使用嵌套列表解析的示例代码：

def flatten_2d_list(lst):
    return [elem for sublist in lst for elem in sublist if isinstance(sublist, list)]

这种方法使用 isinstance() 函数来检查子列表是否是一个列表。如果子列表是一个列表，则它会被展开。

实际案例

假设我们有以下的二维列表：

my_list = [[1, 2, 3], [4, 5, [6, 7]], 8, [9, [10]]]

我们可以使用上面提到的方法中的任何一种来扁平化列表：

flattened_list = flatten_2d_list(my_list)
print(flattened_list)

输出结果为：

[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

总结：本文介绍了几种不同的方法来扁平化二维列表。无论您选择哪种方法，每种方法的目标都是将多个嵌套列表中的元素提取到单个列表中。如果你使用这些方法中的任何一个，你可以方便地使用Python来操作嵌套列表。