Python 进阶小技巧:7招展开嵌套列表

大家好,今天给大家讲解一个Python的进阶知识点:如何将一个嵌套的大列表展开形成一个列表。

小编提供了7种方法供大家学习参考:

  • for循环

  • 列表推导式

  • 使用第三方库itertools

  • 使用sum函数

  • python自加(+=)

  • 使用extend函数

  • 使用递归函数

技术交流

技术要学会分享、交流,不建议闭门造车。一个人走的很快、一堆人可以走的更远。

本文来自技术群粉丝分享整理,文章源码、数据、技术交流,均可加交流群获取,群友已超过2000人,添加时最好的备注方式为:来源+兴趣方向,方便找到志同道合的朋友。

方式①、添加微信号:pythoner666,备注:来自CSDN +备注来意
方式②、微信搜索公众号: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_data  
sum_data = flatten(data)  # 调用递归函数  
sum_data  
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]  

后续

本文的数据有特殊点:嵌套只有两层,且里面的元素全部是列表;后面会考虑非全部列表的数据(比如列表和数字组合的嵌套列表等)和多层嵌套。

如果有更好的方法,留言区欢迎讨论交流~

猜你喜欢

转载自blog.csdn.net/qq_34160248/article/details/129311149