python中的itertools的使用详解


Posted in Python onJanuary 13, 2020

今天了解了下python中内置模块itertools的使用,熟悉下,看能不能以后少写几个for,嘿嘿?。

1.无穷的迭代器

1.1 count(start,[step])

count()接受两个参数

  • start:循环开始的数字
  • step:循环中的间隔
from itertools import count

"""
无穷的迭代器 count()
"""
c = count(0, 2)
v = next(c)
while v < 10:
  v = next(c)
  print(v, end=',')

1.2 cycle()

cycle就是一while True,无限循环里面的数字。

"""
无穷迭代器 cycle()
"""
from itertools import cycle

c = cycle('ABCD')
for i in range(10):
  print(next(c), end=',')

1.3 repeat(elem,[n])

重复迭代elem,n次

"""
无穷迭代器 repeat()
"""
from itertools import repeat

r = repeat(1, 3)
for i in range(3):
  print(next(r), end=',')

2. 迭代器

2.1 accumulate(p,[func])

使用func的函数对迭代对象p进行累积。

"""
迭代器 accumulate()
"""
from itertools import accumulate

test_list = [i for i in range(1, 11)]
for i in accumulate(test_list): # 默认是operator.add
  print(i, end=',')
print()
for i in accumulate(test_list, lambda x, y: x * y): # operator.mul
  print(i, end=',')

2.2 chain()

chain()中可以放多个迭代对象,然后一一迭代出来。

"""
迭代器 chain()
"""
from itertools import chain

ch = chain([1, 2, 3], {4: 4, 5: 5}, {6, 7, 8}, (9,), [10, [11, 12]])
for i in ch:
  print(i)

2.3 chain.from_iterable()

跟chain不同的地方在于:

  • chain: 可以接受多个迭代对象
  • chain.from_iterable():可以接受一个可以产生迭代对象的迭代器
"""
迭代器 chain.from_iterable()
"""
def gen_iterables():
  for i in range(10):
    yield range(i)

for i in chain.from_iterable(gen_iterables()):
  print(i)

2.4 compress(data,selectors)

这是就是看下这个就知道了s是selectors中的元素。
(d[0] if s[0]), (d[1] if s[1]), ...

"""
迭代器 compress
"""
from itertools import compress

print(list(compress(['A', 'B', 'C', 'D'], [0, 1, 1, 1])))

2.5 dropwhile(pred,seq)

循环开始的条件是,直到遇到第一次不满足pred条件的情况,才开始遍历。

"""
迭代器 dropwhile()
"""
from itertools import dropwhile

l = [1, 7, 6, 3, 8, 2, 10]
print(list(dropwhile(lambda x: x < 3, l)))

2.6 groupby

这个感觉挺有意思的,有点像sql中的group_by。可以对字符串,列表等进行分组。

返回键和,组里的内容

from itertools import groupby

# 对字符串进行分组
for k, g in groupby('11111234567'):
  print(k, list(g))
d = {1: 1, 2: 2, 3: 2}
# 按照字典value来进行分组
for k, g in groupby(d, lambda x: d.get(x)):
  print(k, list(g))

2.7 islice
这个就是对迭代对象进行切割,不支持负数,有点像range(1,10,2)这种

from itertools import islice
print(list(islice('ABCDEFG', 2,3, None)))

2.8 zip_longest

这个和zip很像,不同地方在于:

  • zip结束取决于里面最短的迭代对象
  • zip_longest结束取决于里面最长的迭代对象
from itertools import zip_longest

for x,y in zip_longest([1,2,3],[1,2]):
  print(x,y)
for x,y in zip([1,2,3],[1,2]):
  print(x,y)

排列组合迭代器

3.1 product

相当于 嵌套的for

“”"
排列组合迭代器 product 嵌套的for
“”"
from itertools import product
for i,j in product([1,2,3],[4,5]):
print(i,j

3.2 permutations

全排列,比如输出123的全部情况。(1,2,3),(1,3,2)…

from itertools import permutations
print(list(permutations('123')))

3.3 combinations(p,r)

从p中找出所有长度为r的排列情况… 有顺序

from itertools import combinations
print(list(combinations([1,2,3],2)))

3.4 combinations_with_replacement()

从p中找出所有长度为r的排列情况,有顺序,但包括自身就是会重复的意思。

  • combinations_with_replacement(‘ABCD', 2)
  • AA AB AC AD BB BC BD CC CD DD

了解是了解了,就是用的时候不知道能不能想起来…

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python采用getopt解析命令行输入参数实例
Sep 30 Python
Python合并两个字典的常用方法与效率比较
Jun 17 Python
python递归删除指定目录及其所有内容的方法
Jan 13 Python
Python中%r和%s的详解及区别
Mar 16 Python
Python用sndhdr模块识别音频格式详解
Jan 11 Python
使用Python进行QQ批量登录的实例代码
Jun 11 Python
python+opencv实现阈值分割
Dec 26 Python
python3实现指定目录下文件sha256及文件大小统计
Feb 25 Python
详解numpy矩阵的创建与数据类型
Oct 18 Python
pytorch 实现cross entropy损失函数计算方式
Jan 02 Python
Python多线程获取返回值代码实例
Feb 17 Python
利用Python实现字幕挂载(把字幕文件与视频合并)思路详解
Oct 21 Python
python3读取csv文件任意行列代码实例
Jan 13 #Python
pytorch程序异常后删除占用的显存操作
Jan 13 #Python
Python跑循环时内存泄露的解决方法
Jan 13 #Python
PyTorch使用cpu加载模型运算方式
Jan 13 #Python
Python如何读取文件中图片格式
Jan 13 #Python
详解python破解zip文件密码的方法
Jan 13 #Python
PyTorch 随机数生成占用 CPU 过高的解决方法
Jan 13 #Python
You might like
MySql中正则表达式的使用方法描述
2008/07/30 PHP
同台服务器使用缓存APC效率高于Memcached的演示代码
2010/02/16 PHP
php生成的html meta和link标记在body标签里 顶部有个空行
2010/05/18 PHP
PHP数组排序函数合集 以及它们之间的联系分析
2013/06/27 PHP
PHP+MySql+jQuery实现的&quot;顶&quot;和&quot;踩&quot;投票功能
2016/05/21 PHP
Ctrl+Enter提交内容信息
2006/06/26 Javascript
对YUI扩展的Gird组件 Part-1
2007/03/10 Javascript
IE6下js通过css隐藏select的一个bug
2010/08/16 Javascript
通过Jquery遍历Json的两种数据结构的实现代码
2011/01/19 Javascript
jquery向.ashx文件post中文乱码问题的解决方法
2011/03/28 Javascript
js中的string.format函数代码
2020/08/11 Javascript
客户端验证用户名和密码的方法详解
2016/06/16 Javascript
JavaScript中Promise的使用详解
2017/02/26 Javascript
JavaScript 过滤关键字
2017/03/20 Javascript
微信小程序教程系列之视图层的条件渲染(10)
2017/04/19 Javascript
Webpack打包css后z-index被重新计算的解决方法
2017/06/18 Javascript
VueJs组件之父子通讯的方式
2018/05/06 Javascript
vue自定义一个v-model的实现代码
2018/06/21 Javascript
JavaScript数组方法的错误使用例子
2018/09/13 Javascript
使用Object.defineProperty如何巧妙找到修改某个变量的准确代码位置
2018/11/02 Javascript
微信小程序时间轴实现方法示例
2019/01/14 Javascript
小程序server请求微信服务器超时的解决方法
2019/05/21 Javascript
axios如何利用promise无痛刷新token的实现方法
2019/08/27 Javascript
在vue-cli中引入lodash.js并使用详解
2019/11/13 Javascript
JS XMLHttpRequest原理与使用方法深入详解
2020/04/30 Javascript
2020京东618叠蛋糕js脚本(亲测好用)
2020/06/02 Javascript
Ajax获取node服务器数据的完整步骤
2020/09/20 Javascript
Python global全局变量函数详解
2018/09/18 Python
python连接打印机实现打印文档、图片、pdf文件等功能
2020/02/07 Python
Dockers鞋官网:Dockers Shoes
2018/11/13 全球购物
莫斯科珠宝厂官方网站:Miuz
2020/09/19 全球购物
急诊科护士自我鉴定
2013/10/14 职场文书
党的群众路线教育实践活动总结报告
2014/07/03 职场文书
房屋产权证明书
2014/10/15 职场文书
联欢会开场白
2015/06/01 职场文书
浅谈Redis主从复制以及主从复制原理
2021/05/29 Redis