python itertools模块

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

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

列表变成生成器

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

a = [1,2,3]
print(type(a))
b = iter(a)
print(type(b))

输出结果:

<class 'list'>
<class 'list_iterator'>

获取结果一样的,但是可以使用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)

返回结果:

<class 'itertools.accumulate'>
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)]

参考链接

坚持原创技术分享,您的支持将鼓励我继续创作!
------ 本文结束 ------

版权声明

LangZi_Blog's by Jy Xie is licensed under a Creative Commons BY-NC-ND 4.0 International License
由浪子LangZi创作并维护的Langzi_Blog's博客采用创作共用保留署名-非商业-禁止演绎4.0国际许可证
本文首发于Langzi_Blog's 博客( http://langzi.fun ),版权所有,侵权必究。

0%