Python进阶知识点:如何将一个嵌套的大列表展开形成一个列表

2023-07-0214:11:55编程语言入门到精通Comments992 views字数 1745阅读模式

讲解一个Python的进阶知识点:如何将一个嵌套的大列表展开形成一个列表。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/ymba/49260.html

我提供了7种方法供大家学习参考:文章源自菜鸟学院-https://www.cainiaoxueyuan.com/ymba/49260.html

  • for循环
  • 列表推导式
  • 使用第三方库itertools
  • 使用sum函数
  • python自加(+=)
  • 使用extend函数
  • 使用递归函数

模拟数据

data = [[1,2,3],[4],[5,6,7],[8,9],[10]]  # 模拟数据
data
[[1, 2, 3], [4], [5, 6, 7], [8, 9], [10]]

这份模拟的数据有2个特点:文章源自菜鸟学院-https://www.cainiaoxueyuan.com/ymba/49260.html

  • 嵌套列表只有两层
  • 里面的元素也全部是列表类型

方式1:for循环

最容易想到的方式就是for循环:文章源自菜鸟学院-https://www.cainiaoxueyuan.com/ymba/49260.html

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循环能够实现,那么列表推导式肯定也可以:文章源自菜鸟学院-https://www.cainiaoxueyuan.com/ymba/49260.html

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:文章源自菜鸟学院-https://www.cainiaoxueyuan.com/ymba/49260.html

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函数,给个案例文章源自菜鸟学院-https://www.cainiaoxueyuan.com/ymba/49260.html

# 如何理解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_data
sum_data = flatten(data)  # 调用递归函数
sum_data
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

后续

本文的数据有特殊点:嵌套只有两层,且里面的元素全部是列表;后面会考虑非全部列表的数据(比如列表和数字组合的嵌套列表等)和多层嵌套。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/ymba/49260.html

  • 本站内容整理自互联网,仅提供信息存储空间服务,以方便学习之用。如对文章、图片、字体等版权有疑问,请在下方留言,管理员看到后,将第一时间进行处理。
  • 转载请务必保留本文链接:https://www.cainiaoxueyuan.com/ymba/49260.html

Comment

匿名网友 填写信息

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen:

确定