Python中的迭代工具包 - itertools

你的名字 2022-12-01 14:23 242阅读 0赞

# itertools #

> 是python中的一个工具包, 其中的方法来帮助你生成各种各样的 iterator , 化繁为简, 下面是我的一个简单梳理, 具体解释点下面官方文档
> 
> ###### 官方文档: [Python - itertools][] ######
> 
> ###### 方法的中文名都不是官方定义的, 是我自己取的, 要是有更好的名称可以评论告诉我 ######

from itertools import *

#### 1. 累加 (accumulate) ####

传入iterable , 将其中每个元素累加起来

list(accumulate([1,2,3,4])
    # 输出 [1, 3, 6, 10]
    # 用 next() 依次输出 1, 3, 6, 10

#### 2. 链接 (chain) ####

传入iterable , 将其中每个 iterable 链接起来

a = ['123','456']
    b = ['789','abc']
    print(f'对于多个iter: {list(chain(a,b))}')
    print(f'对于单个iter: {list(chain(*a,*b))}')
    
    print(f'指明参数为iter:{list(chain.from_iterable(a))}')
    # 输出
    # 对于多个iter: ['123', '456', '789', 'abc']
    # 对于单个iter: ['1', '2', '3', '4', '5', '6', '7', '8', '9', 'a', 'b', 'c']
    # 指明参数为iter:['1', '2', '3', '4', '5', '6']

#### 3. 组合 (combination & combinations\_with\_replacement) ####

传入iterable , 按照第二个参数进行组合

# 组合器 不允许重复
    print(f"字母不重复:{list(combinations('ABC',2))}")
    print(f"range方法:{list(combinations(range(3),2))}")
    
    # 组合替换 允许重复
    print(f"字母重复:{list(combinations_with_replacement('ABC',2))}")
    
    # 输出
    # 字母不重复:[('A', 'B'), ('A', 'C'), ('B', 'C')]
    # range方法:[(0, 1), (0, 2), (1, 2)]
    # 字母重复:[('A', 'A'), ('A', 'B'), ('A', 'C'), ('B', 'B'), ('B', 'C'), ('C', 'C')]

#### 4. 排列 (permutation) ####

*  对 iterable 按照 k 的数量进行排列

list(permutations(range(3),2))
    
    # 输出
    # [(0, 1), (0, 2), (1, 0), (1, 2), (2, 0), (2, 1)]

#### 5. 匹配 (compress) ####

*  传入 iterable , 将满足第二个参数的中为true的输出

print(f"按照第二个iter匹配,为1的显示,0不显示:{list(compress('abcde',[0,1,1,0,1]))}")
    
    # 输出
    # 按照第二个iter匹配,为1的显示,0不显示:['b', 'c', 'e']

#### 6. 计数 (count) ####

*  传入start, step , 将在start的基础上进行计数输出

coun = count(10,3)
    print(next(coun),next(coun),next(coun))
    # 输出
    # 10,11,12

#### 7. 周期循环 (cycle) ####

*  对传入的 iterable 进行无限循环

cycl = cycle('abc')
    print(next(cycl),next(cycl),next(cycl),next(cycl),next(cycl),next(cycl))
    # 输出
    # a b c a b c

#### 8. 重复 (repeat) ####

*  对传入的 任何值 进行 k 次循环

list(repeat(4,3))
    # 输出
    # 4,4,4

#### 9. 忽略循环 (takewhile & dropwhile) ####

*  takewhile 不满足条件则停止迭代, 从第一个满足条件的值开始输出
 *  dropwhile 满足条件则跳过, 从第一个不满足条件的值开始输出

# 满足条件则跳过, 从第一个不满足条件的值开始输出
    dw = dropwhile(lambda x: x<5, [1, 4, 6, 4, 1])
    print('满足条件则跳过, 从第一个不满足条件的值开始输出:',next(dw),next(dw),next(dw))
    
    # 不满足条件则停止迭代, 从第一个满足条件的值开始输出
    tw = takewhile(lambda x: x<5, [1, 4, 6, 4, 1])
    print('不满足条件则停止迭代, 从第一个满足条件的值开始输出:',next(tw),next(tw))
    # 输出
    # 满足条件则跳过, 从第一个不满足条件的值开始输出: 6 4 1
    # 不满足条件则跳过, 从第一个满足条件的值开始输出: 1 4

#### 10. 过滤 (filter& filterfalse) ####

*  filter 过滤符合条件的值
 *  dropwhile 过滤不符合条件的值

filt = filter(lambda x:x%2,range(10))
    print('符合条件的值过滤:',next(filt),next(filt),next(filt),next(filt))
    
    ff = filterfalse(lambda x:x%2,range(10))
    print('不符合条件的值过滤:',next(ff),next(ff),next(ff),next(ff))
    # 输出
    # 符合条件的值过滤: 1 3 5 7
    # 不符合条件的值过滤: 0 2 4 6

#### 11. 分组 (grouby) ####

*  对 iterable 进行分组, 具体解释看官网吧, 我就把例子放在这了

print(list([k for k, g in groupby('AAAABBBCCDAABBB')]))
    print(list([list(g) for k, g in groupby('AAAABBBCCD')]))
    # 输出
    # ['A', 'B', 'C', 'D', 'A', 'B']
    # [['A', 'A', 'A', 'A'], ['B', 'B', 'B'], ['C', 'C'], ['D']]

#### 12. 切片 (islice) ####

*  islice(iterable,stop)
 *  islice(iterable,start,stop,\[step\])

list(islice('ABCD',2))
    # 输出
    # ['A', 'B']

#### 13. 笛卡尔积 (product) ####

list(product('abc','xy'))
    # 输出
    # [('a', 'x'), ('a', 'y'), ('b', 'x'), ('b', 'y'), ('c', 'x'), ('c', 'y')]

#### 14. 星图 (startmap) ####

*  startmap(func, iterable)
 *  对 iterable 按照 func 的处理方式进行处理

list(starmap(pow,[[2,2],(2,5),(4,2)]))
    # 输出
    # [4.0, 32.0, 16.0]

#### 15. 分裂 (tee) ####

*  将一个可迭代对象分裂成多个可迭代对象, 就像细胞分裂一样
 *  tee(iterable , n)

tee([1,2],2)
    # 按照 [1,2] 生成 2 个 [1,2]

#### 16. zip补齐 (zip\_longest) ####

*  原始zip任意一方超出则截止, zip\_longest使用 fillvalue 将超出的部分补齐
 *  zip(iterable , ,iterable, fullvalue)

print("原始zip:",list(zip([1,2,3],['a','b'])))
    print("zip_longest:",list(zip_longest([1,2,3],['a','b'],fillvalue='补齐')))
    # 输出
    # 原始zip: [(1, 'a'), (2, 'b')]
    # zip_longest: [(1, 'a'), (2, 'b'), (3, '补齐')]

# Just it #

[Python - itertools]: https://docs.python.org/zh-cn/3.7/library/itertools.html#itertools.chain