python itertools模块

在数据处理中还比较常见，整理了一下。内容主要来自 python 官方文档，更多详细，请参考这里。

这是一个python内建的模块，主要用于实现迭代。灵感来源于APL、Haskell and SML等语言。

特点：以纯python的方式来实现，速度较快。

无限迭代函数，主要有三个

iterator	arguments	results	example
count()	start, [step]	start, start+step, start+2*step, …	count(10) --> 10 11 12 13 14 …
cycle()	p	p0, p1, … plast, p0, p1, …	cycle(‘ABCD’) --> A B C D A B C D …
repeat()	elem [,n]	elem, elem, elem, … endlessly or up to n times	repeat(10, 3) --> 10 10 10

非无限迭代

函数较多。比如accumulate() 返回累加的值的迭代，

组合迭代

比较有趣，主要有四个

Iterator	Arguments	Results
product()	p, q, … [repeat=1]	cartesian product, equivalent to a nested for-loop类似于做笛卡尔积
permutations()	p[, r]	r-length tuples, all possible orderings, no repeated elements r个元素的全排列
combinations()	p, r	r-length tuples, in sorted order, no repeated elements r个元素的组合排列
combinations_with_replacement()	p, r	r-length tuples, in sorted order, with repeated elements 允许放回的组合排列
product(‘ABCD’, repeat=2)		AA AB AC AD BA BB BC BD CA CB CC CD DA DB DC DD
permutations(‘ABCD’, 2)		AB AC AD BA BC BD CA CB CD DA DB DC
combinations(‘ABCD’, 2)		AB AC AD BC BD CD
combinations_with_replacement(‘ABCD’, 2)		AA AB AC AD BB BC BD CC CD DD

应用示例

求质数序列中1,3,5,7,9,11,13,15三个数之和为35的三个数；

def get_three_data(data_list,amount):
    for data in list(itertools.combinations(data_list, 3)):
        if sum(data) == amount:
            print(data)
#(7, 13, 15)
#(9, 11, 15)