首页 > 代码库 > 压平列表
压平列表
碾平列表是个很好玩的函数。比如你有个嗷嗷恶心的列表：
[[1, 2], [3, [4], [5, 6], 7], 8]
你想把它变成正常一点的
[1, 2, 3, 4, 5, 6, 7, 8]
要怎么办呢？
老实说，很久不接触这种东西，我已经早忘了当初是怎么写的了，憋了半天没写出来，后来参考了 http://www.cnblogs.com/c-hy/archive/2012/09/21/2696703.html 才回忆起来。
然后，着重介绍三种方法：
1、使用sum和map，感觉玩的比较炫，大家来感受一下：
from collections import Iterabledef flatten(x): if isinstance(x, Iterable) and not isinstance(x, (str, bytes)): return sum(map(flatten, x), []) else: return [x]lst = [1, 2, [3, [4], [5, 6], 7], 8]print(flatten(lst))
刚才那个网页中说，国外某论坛的大神写了一个匿名函数
flat=lambda L: sum(map(flat,L),[]) if isinstance(L,list) else [L]
基本上是一个意思的。
2、使用yield。这个是在《Python Cookbook》中介绍的一种方法
from collections import Iterabledef flatten(items, ignore_types=(str, bytes)): for x in items: if isinstance(x, Iterable) and not isinstance(x, ignore_types): yield from flatten(x) else: yield xitems = [[1, 2], [3, [4], [5, 6], 7], 8]# Produces 1 2 3 4 5 6 7 8for x in flatten(items): print(x)
yield from 比较巧妙。不过，这里有个限制，yield from是python3才有的，如果是在python2中使用，需要这么写：
from collections import Iterabledef flatten(items, ignore_types=(str, bytes)): for x in items: if isinstance(x, Iterable) and not isinstance(x, ignore_types): for i in flatten(x): yield i else: yield xitems = [[1, 2], [3, [4], [5, 6], 7], 8]# Produces 1 2 3 4 5 6 7 8for x in flatten(items): print(x)
2和3通用。不得不说yield真是个好东西。
3、Tkinter自带函数_flatten。不说话，直接上代码
from Tkinter import _flattenlst = [1, 2, [3, [4], [5, 6], 7], 8]print(_flatten(lst))
这个就比较可怕了。不过Tkinter这东西，据说只有windows才有，linux没有。mac不太清楚。
当然，其他方法也是有的，比如
def flatten(x): for i in x: if isinstance(i, Iterable) and not isinstance(i, (str, bytes)): flatten(i) else: new_lst.append(i)new_lst = []lst = [1, 2, [3, [4], [5, 6], 7], 8]flatten(lst)print(new_lst)
（感谢群友提供）思路比较简单，代码也很清晰，只是全局列表到底还是感觉有一点点耍赖。
当然，要说耍赖，这里还有更耍赖的，比如
lst = [1, 2, [3, [4], [5, 6], 7], 8]print(list("[" + str(lst).replace("[", "").replace("]", "") + "]"))
还有
import relst = [1, 2, [3, [4], [5, 6], 7], 8]print map(int, re.findall(‘\d+‘, str(lst)))
哈哈，很特定的环境下可以使用，比如第一个列表元素中不能含有"["和"]"，第二个更是只能包含整数。不过，同学们的脑洞还是比较大的。
记录一下，留着玩。
压平列表
声明：以上内容来自用户投稿及互联网公开渠道收集整理发布，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任，若内容有误或涉及侵权可进行投诉： 投诉/举报 工作人员会在5个工作日内联系你，一经查实，本站将立刻删除涉嫌侵权内容。