itertools库
迭代器(生成器)在Python中是一种很常用也很好用的数据结构,比起列表(list)来说,迭代器最大的优势就是延迟计算,按需使用,从而提高开发体验和运行效率,以至于在Python 3中map,filter等操作返回的不再是列表而是迭代器。
话虽这么说但大家平时用到的迭代器大概只有range了,而通过iter函数把列表对象转化为迭代器对象又有点多此一举,这时候我们今天的主角itertools就该上场了。
使用itertools
itertools中的函数大多是返回各种迭代器对象,其中很多函数的作用我们平时要写很多代码才能达到,而在运行效率上反而更低,毕竟人家是系统库。
itertools.accumulate
简单来说就是累加。
itertools.chain
连接多个列表或者迭代器。
itertools.combinations
求列表或生成器中指定数目的元素不重复的所有组合
itertools.combinations_with_replacement
允许重复元素的组合
itertools.compress
按照真值表筛选元素
itertools.count
就是一个计数器,可以指定起始位置和步长
itertools.cycle
循环指定的列表和迭代器
itertools.dropwhile
按照真值函数丢弃掉列表和迭代器前面的元素
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,不过已较长的列表和迭代器的长度为准