这段代码很Pythonic：相见恨晚的itertools库

前言

最近事情不是很多，想写一些技术文章分享给大家，同时也对自己一段时间来碎片化接受的知识进行一下梳理，所谓写清楚才能说清楚，说清楚才能想清楚，就是这个道理了。

很多人都致力于把Python代码写得更Pythonic，一来更符合规范且容易阅读，二来一般Pythonic的代码在执行上也更有效率。今天就先给大家介绍一下Python的系统库itertools。

itertools库

迭代器（生成器）在Python中是一种很常用也很好用的数据结构，比起列表(list)来说，迭代器最大的优势就是延迟计算，按需使用，从而提高开发体验和运行效率，以至于在Python 3中map,filter等操作返回的不再是列表而是迭代器。

话虽这么说但大家平时用到的迭代器大概只有range了，而通过iter函数把列表对象转化为迭代器对象又有点多此一举，这时候我们今天的主角itertools就该上场了。

使用itertools

itertools中的函数大多是返回各种迭代器对象，其中很多函数的作用我们平时要写很多代码才能达到，而在运行效率上反而更低，毕竟人家是系统库。

itertools.accumulate

简单来说就是累加。

itertools.chain

连接多个列表或者迭代器。

itertools.combinations

求列表或生成器中指定数目的元素不重复的所有组合

itertools.compress

按照真值表筛选元素

就是一个计数器,可以指定起始位置和步长

itertools.cycle

循环指定的列表和迭代器

按照真值函数丢弃掉列表和迭代器前面的元素

itertools.filterfalse

保留对应真值为False的元素

itertools.groupby

按照分组函数的值对元素进行分组

itertools.islice

上文使用过的函数，对迭代器进行切片

itertools.permutations

产生指定数目的元素的所有排列(顺序有关)

itertools.product

产生多个列表和迭代器的(积)

itertools.repeat

简单的生成一个拥有指定数目元素的迭代器

itertools.starmap

类似map

itertools.takewhile

与dropwhile相反，保留元素直至真值函数值为假。

itertools.tee

这个函数我也不是很懂，似乎是生成指定数目的迭代器

itertools.zip_longest

类似于zip，不过已较长的列表和迭代器的长度为准

结语

大概就总结到这里，不过老实说Python的各种语言特性和库还是要多用才能熟练，最终达到随手拈来的程度，装逼的说就是由术入道。

原文：https://segmentfault.com/a/1190000008590958

学习Python和网络爬虫，关注公众号：datanami