python itertools模块_python itertools模块

每个人心里都有一个魔鬼，幸福是他的牢笼，当人们的幻想化作泡影，恶魔将唱着血腥的圣歌降临，那时绝望的人将所向无敌！

生成器相对使用列表保存数据来说，优势在于可以延时计算，按需使用，从而提高开发体验的运行效率，在python3中，map，filter返回的不再是列表而是生成器。

列表变成生成器

使用iter将列表转换成生成器

a = [1,2,3]

print(type(a))

b = iter(a)

print(type(b))

输出结果：

获取结果一样的，但是可以使用next()方法依次获取，如果获取过一次后就不能再获取第二次了，

# for x in b:

# print(x)

print(next(b))

print(next(b))

print(next(b))

itertools.repeat

生成指定数目的生成器

a = itertools.repeat(1,5)

print(list(a))

返回结果：

[1, 1, 1, 1, 1]

itertools.accumlate

数据累加

import itertools

a = range(100)

b = (itertools.accumulate(a))

print(type(b))

for x in b:

print(x)

返回结果：

0，1，3.....4950

itertools.product

上面的是获取相加的结果，下面的是获取相乘的结果

a = itertools.product('浪子', '好帅')

print(list(a))

返回结果：

[('浪', '好'), ('浪', '帅'), ('子', '好'), ('子', '帅')]

当然除了字符串还可以是列表。

itertools.chain

列表合并，注意合并的结果会去除重复

a = range(10)

b = range(20)

c = itertools.chain(a,b)

print(list(c))

返回结果：

[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19]

itertools.count

获取一个无限长的数字，看看例子就明白了，因为打印速度太快所以暂停2s

a = itertools.count(10,1)

for x in a:

print(x)

time.sleep(2)

返回结果:

10

11

12

13

打印到13的时候我按了结束，意思就是从10开始，每个递增1，当然也可以是负数。

itertools.cycle

无限逐个循环列表的数据内容，可以用来把一些函数放在一个列表，然后做无限循环执行~

a = itertools.cycle([1,2,3])

for x in a:

print(x)

返回结果就是不断的显示1，2，3，1，2，3……….

itertools.compress

按照自己的要求过滤掉列表数组的内容，比较傻瓜化

b = itertools.compress([1,2,3,4,5],(True,True,False,False,False))

print(list(b))

返回结果：

[1, 2]

当然你可以使用0，1来代替True和False

a = itertools.compress([1,2,3,4,5],(0,0,0,1,1))

print(list(a))

返回结果：

[4, 5]

感觉这个的话，同样可以用在扫描器中，根据扫描函数返回的True和False在提前设定要想要的相关数据，可以简化很多代码，但是易读性的话如果没接触这个库的小伙伴可能不明白是怎么回事

itertools.dropwhile

作用是丢弃删除你的过滤规则中符合规则的

zz = itertools.dropwhile(lambda a:a<5,range(10))

print(list(zz))

返回结果：

[5, 6, 7, 8, 9]

itertools.takewhile

和上面的用法完全逸雅阁，但实际保留的是符合要求的结果

itertools.filterfalse

和上面的差不多的原理，意思就是排除表达式中为否的数字

zz = itertools.filterfalse(lambda a:a<5,range(10))

print(list(zz))

返回结果：

[5, 6, 7, 8, 9]

itertools.groupby

和上面的作用也很像，根据你的表达式返回的结果进行分类。

a = itertools.groupby([1,0,2,3,4,5,6],lambda x:x>5)

for x,y in a:

print(x,'::::',list(y))

返回结果：

False :::: [1, 0, 2, 3, 4, 5]

True :::: [6]

itertools.islice

和列表的切片很相似

a = itertools.islice([0,0,2,2,4,4,5,5],4)

# 获取前4为，就是0，4

print(list(a))

a = itertools.islice([0,0,2,2,4,4,5,5],0,2)

# 获取前2为，就是0，2

print(list(a))

a = itertools.islice([0,0,2,2,4,4,5,5],3,6)

# 获取前3-6位

print(list(a))

a = itertools.islice([0,0,2,2,4,4,5,5],0,5,2)

# 获取前0-5位，并且间隔位2

print(list(a))

返回结果：

[0, 0, 2, 2]

[0, 0]

[2, 4, 4]

[0, 2, 4]

itertools.zip_longest

上面那个和列表的切片很相似，这个就和zip的作用很相似

>>> x = itertools.zip_longest(range(3), range(5))

>>> y = zip(range(3), range(5))

>>> print(list(x))

[(0, 0), (1, 1), (2, 2), (None, 3), (None, 4)]

>>> print(list(y))

[(0, 0), (1, 1), (2, 2)]

itertools.starmap

上面那个和zip很像，这个就是和map很像了，根据函数返回的结果生成一个列表内容

import itertools

import string

def test(x):

if x in string.ascii_lowercase:

return True

else:

return False

a = itertools.starmap(test,['a','A','1'])

print(list(a))

返回结果：

[True, False, False]

itertools.combinations

在一个列表中获取指定的数组，并且不能重复，举一个例子：

a = itertools.combinations([1,2,3,4,5,6],3)

for x in a:

print(x)

是从列表1-6中取3个值组成一个数组，并且不能重复

返回结果：

(1, 2, 3)

(1, 2, 4)

(1, 2, 5)

(1, 2, 6)

(1, 3, 4)

(1, 3, 5)

(1, 3, 6)

(1, 4, 5)

(1, 4, 6)

(1, 5, 6)

(2, 3, 4)

(2, 3, 5)

(2, 3, 6)

(2, 4, 5)

(2, 4, 6)

(2, 5, 6)

(3, 4, 5)

(3, 4, 6)

(3, 5, 6)

(4, 5, 6)

itertools.combinations_with_replacement

和上面的一样，但是运行数据可以重复

itertools.permutations

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

x = itertools.permutations(range(4), 3)

print(list(x))

返回结果：

[(0, 1, 2), (0, 1, 3), (0, 2, 1), (0, 2, 3), (0, 3, 1), (0, 3, 2), (1, 0, 2), (1, 0, 3), (1, 2, 0), (1, 2, 3), (1, 3, 0), (1, 3, 2), (2, 0, 1), (2, 0,3), (2, 1, 0), (2, 1, 3), (2, 3, 0), (2, 3, 1), (3, 0, 1), (3, 0, 2), (3, 1, 0), (3, 1, 2), (3, 2, 0), (3, 2, 1)]