python进阶-嵌套列表展开
python进阶-嵌套列表展开
皮大大
发布于 2023-08-25 04:08:54
发布于 2023-08-25 04:08:54
代码可运行
运行总次数：0
代码可运行
本文讲解的是一个Python的进阶知识点：**如何将一个嵌套的大列表展开形成一个大列表。
模拟数据
data = [[1,2,3],[4],[5,6,7],[8,9],[10]] # 模拟数据
data[[1, 2, 3], [4], [5, 6, 7], [8, 9], [10]]这份模拟的数据有2个特点：
- 嵌套列表只有两层
- 里面的元素也全部是列表类型
方式1：for循环
最容易想到的方式就是for循环：
for i in data:
print(type(i))<class 'list'>
<class 'list'>
<class 'list'>
<class 'list'>
<class 'list'># 导入Iterable 模块
from collections import Iterable
sum_data = []
for i in data:
if isinstance(i,Iterable): # 如果可迭代（比如列表形式）
for j in i: # 再次循环追加元素
sum_data.append(j)
else:
sum_data.append(i) # 否则直接追加
sum_data[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]方式2：列表推导式
for循环能够实现，那么列表推导式肯定也可以：
sum_data = [i for j in data if isinstance(j,Iterable) for i in j] # 循环一定是从大到小
sum_data[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]方式3：使用itertools库
借助第三方的库itertools：
import itertools
# 通过chain方法从可迭代对象中生成；最后展开成列表
sum_data = list(itertools.chain.from_iterable(data))
sum_data[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]方式4：使用sum函数
sum_data = sum(data, [])
sum_data[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]方式5：Python自加
sum_data = []
for i in data:
sum_data += i # 实现自加
sum_data[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]方式6：extend函数
如何快速理解python的extend函数，给个案例
# 如何理解python的extend函数
list1 = [1,2,3,4]
list1.extend([5,6]) # 追加功能extend；就地修改
list1 # list1已经发生了变化[1, 2, 3, 4, 5, 6]sum_data = []
for i in data:
sum_data.extend(i) # 对空列表逐步追加
sum_data[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]方式7：使用递归函数
data[[1, 2, 3], [4], [5, 6, 7], [8, 9], [10]]def flatten(data): # 定义递归函数
sum_data = []
for i in data:
if isinstance(i, Iterable): # 如果i是可迭代的对象（列表等），调用函数本身；直到执行else语句
sum_data.extend(flatten(i))
else:
sum_data.append(i)
return sum_datasum_data = flatten(data) # 调用递归函数
sum_data[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]后续
本文的数据有特殊点：嵌套只有两层，且里面的元素全部是列表；后面会考虑非全部列表的数据（比如列表和数字）和多层嵌套。
本文参与 腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。
原始发表：2023-3-1，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读
目录
腾讯云开发者
