前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >python技巧(2)--碾平列表和列表去重

python技巧(2)--碾平列表和列表去重

作者头像
kbsc13
发布2020-02-19 09:44:42
1.4K0
发布2020-02-19 09:44:42
举报
文章被收录于专栏:AI 算法笔记AI 算法笔记

总第 116 篇文章,本文大约 1000 字,阅读大约需要 3 分钟

今天介绍和列表相关的两个小技巧:

  • 碾平列表(flatten list),也就是列表里的元素也带有列表的情况;
  • 列表去重,保留原始顺序和不保留顺序的做法

1. 碾平列表

碾平列表(flatten list ),即当列表里面嵌套列表,如何将这些子列表给取出来,得到一个不包含子列表的列表,示例如下:

代码语言:javascript
复制
list1 = [1, [2, [3,4]], 5]

=>new_list = [1, 2, 3, 4, 5]

这里介绍 3 种方法,分别如下。

方法1:利用递归的思想,代码如下:

代码语言:javascript
复制
list1 = [1, [2, [3,4]], 5]
res = []

def fun(s):
    for i in s:
        if isinstance(i, list):
            fun(i)
        else:
            res.append(i)

fun(list1)
print(res)

接着是两种比较高级的写法,用 lambda 实现一个匿名函数

方法2:

代码语言:javascript
复制
flat = lambda L: sum(map(flat, L), []) if isinstance(L, list) else [L]

print(flat(list1))

方法3:

代码语言:javascript
复制
a = [1, 2, [3, 4], [[5, 6], [7, 8]]]

flatten = lambda x: [y for l in x for y in flatten(l)] if type(x) is list else [x]

print(flatten(a))

2. 列表去重

列表去重可能会破坏原有的顺序,所以下面分别介绍保留顺序和不保留顺序的做法。

去重,但改变顺序

去重但改变顺序,两种方法

方法1 就是利用 set 进行去重

代码语言:javascript
复制
l1 = ['b','c','d','b','c','a','a']
l2 = list(set(l1))
print l2

方法2 是利用字典的键不重复的特性,将列表的元素作为一个字典的键,然后返回这个字典的所有键,即可实现去重的操作。

代码语言:javascript
复制
l1 = ['b','c','d','b','c','a','a']
l2 = {}.fromkeys(l1).keys()
print l2
去重,不改变顺序

利用 sortedset 方法实现去重并保留原始顺序,这里 sorted 指定排序的规则就是按照原列表的索引顺序

代码语言:javascript
复制
l1 = ['b','c','d','b','c','a','a']
l2 = sorted(set(l1),key=l1.index)
print l2
本文参与 腾讯云自媒体同步曝光计划,分享自微信公众号。
原始发表:2020-02-08,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 算法猿的成长 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 1. 碾平列表
  • 2. 列表去重
    • 去重,但改变顺序
      • 去重,不改变顺序
      领券
      问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档