如何在Python中展平浅列表？

在Python中，有时候会遇到需要将多维的嵌套列表展平成一维的浅列表的情况。比如将多个DataFrame或Series合并为一个DataFrame时，需要将所有数据展平后才能进行操作。那么，如何在Python中展平浅列表呢？本文将介绍3种方法供大家参考。

阅读更多：Python 教程

方法一：使用递归实现展平

递归是一种非常实用的算法，它能够将一个复杂的问题简化为多个相同或类似的子问题，从而简化代码的编写和理解。以下是使用递归实现展平的示例代码：

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

代码解析：

首先定义了一个名为flatten的函数，它接收一个列表lst作为参数。在函数内部，首先定义了一个空列表result，用于存储展平后的数据。然后使用一个for循环遍历传入的列表lst中的每一个元素。

如果当前遍历到的元素是一个列表，则递归调用flatten函数继续展平，最终将展平后的数据存储在result列表中。否则，将当前元素直接加入result列表中。最后返回result列表，即为展平后的浅列表。

示例：

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

方法二：使用列表推导式实现展平

Python中的列表推导式可以帮助我们在一行代码中快速生成一个列表。以下是使用列表推导式实现展平的示例代码：

def flatten(lst):
    return [item for sublist in lst for item in (sublist if isinstance(sublist, list) else [sublist])]

代码解析：

定义了一个名为flatten的函数，它接收一个列表lst作为参数。在函数内部，使用一个列表推导式将传入的多维列表展平。

首先遍历传入的列表lst中的每一个元素，将其中的每一个子列表使用列表推导式展平。如果当前元素不是一个列表，则将其转换为只包含该元素的列表。

示例：

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

方法三：使用itertools库实现展平

Python中的itertools库提供了一些实用的工具函数，例如用于迭代的组合、排列、过滤等函数。其中的chain()函数可以用于将多个可迭代对象合并为一个可迭代对象。以下是使用itertools库实现展平的示例代码：

import itertools

def flatten(lst):
    return list(itertools.chain.from_iterable(lst))

代码解析：

首先导入itertools库，然后定义一个名为flatten的函数，它接收一个列表lst作为参数。在函数内部，使用chain()函数从传入的多维列表中合并所有元素，并用list()函数将其转换为一个浅列表。

示例：

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

结论

展平浅列表在Python编程中是一个常见的任务，本文介绍了3种方法来实现这个目的：使用递归、使用列表推导式和使用itertools库。这些方法各有优缺点，程序员可以根据实际需求选择最适合的方法。无论哪种方法，都可以帮助我们快速地将多维列表展平成一维的浅列表，方便数据操作和处理。