详解python itertools功能


Posted in Python onFebruary 07, 2020

介绍

      itertools是python内置的模块,使用简单且功能强大,这里尝试汇总整理下,并提供简单应用示例;如果还不能满足你的要求,欢迎加入补充。

      使用只需简单一句导入:import itertools

chain()

      与其名称意义一样,给它一个列表如 lists/tuples/iterables,链接在一起;返回iterables对象。

letters = ['a', 'b', 'c', 'd', 'e', 'f']
booleans = [1, 0, 1, 0, 0, 1]
   print(list(itertools.chain(letters,booleans)))
#   ['a', 'b', 'c', 'd', 'e', 'f', 1, 0, 1, 0, 0, 1]
 
  print(tuple(itertools.chain(letters,letters[3:])))
#   ('a', 'b', 'c', 'd', 'e', 'f', 'd', 'e', 'f')
 
  print(set(itertools.chain(letters,letters[3:])))
#   {'a', 'd', 'b', 'e', 'c', 'f'}
    
  print(list(itertools.chain(letters,letters[3:])))
#   ['a', 'b', 'c', 'd', 'e', 'f', 'd', 'e', 'f']
 
  for item in list(itertools.chain(letters,booleans)):
    print(item)

count()

  生成无界限序列,count(start=0, step=1) ,示例从100开始,步长为2,循环10,打印对应值;必须手动break,count()会一直循环。

i = 0
  for item in itertools.count(100,2):
    i += 1
    if i > 10 : break
    
    print(item) 
 
filterfalse ()
   Python filterfalse(contintion,data) 迭代过滤条件为false的数据。如果条件为空,返回data中为false的项;
booleans = [1, 0, 1, 0, 0, 1]
numbers = [23, 20, 44, 32, 7, 12]
 
print(list(itertools.filterfalse(None,booleans)))
#   [0, 0, 0]
print(list(itertools.filterfalse(lambda x : x < 20,numbers)))
#  [23, 20, 44, 32]

compress()

返回我们需要使用的元素,根据b集合中元素真值,返回a集中对应的元素。

print(list(itertools.compress(letters,booleans)))
# ['a', 'c', 'f']

starmap()

      针对list中的每一项,调用函数功能。starmap(func,list[]) ;

starmap(pow, [(2,5), (3,2), (10,3)]) --> 32 9 1000
 
>>> from itertools import *
>>> x = starmap(max,[[5,14,5],[2,34,6],[3,5,2]])
>>> for i in x:
>>> print (i)
14
34
5
repeat()
repeat(object[, times]) 重复times次;
repeat(10, 3) --> 10 10 10
dropwhile()
dropwhile(func, seq );当函数f执行返回假时, 开始迭代序列
dropwhile(lambda x: x<5, [1,4,6,4,1]) --> 6 4 1
takewhile()
takewhile(predicate, iterable);返回序列,当predicate为true是截止。
takewhile(lambda x: x<5, [1,4,6,4,1]) --> 1 4
islice()
islice(seq[, start], stop[, step]);返回序列seq的从start开始到stop结束的步长为step的元素的迭代器
for i in islice("abcdef", 0, 4, 2):#a, c
  print i

product()

product(iter1,iter2, ... iterN, [repeat=1]);创建一个迭代器,生成表示item1,item2等中的项目的笛卡尔积的元组,repeat是一个关键字参数,指定重复生成序列的次数

   

# product('ABCD', 'xy') --> Ax Ay Bx By Cx Cy Dx Dy
  # product(range(2), repeat=3) --> 000 001 010 011 100 101 110 111
for i in product([1, 2, 3], [4, 5], [6, 7]):
  print i
(1, 4, 6)
(1, 4, 7)
(1, 5, 6)
(1, 5, 7)
(2, 4, 6)
(2, 4, 7)
(2, 5, 6)
(2, 5, 7)
(3, 4, 6)
(3, 4, 7)
(3, 5, 6)
(3, 5, 7)

permutations()

permutations(p[,r]);返回p中任意取r个元素做排列的元组的迭代器

for i in permutations([1, 2, 3], 3):
  print i
(1, 2, 3)
(1, 3, 2)
(2, 1, 3)
(2, 3, 1)
(3, 1, 2)
(3, 2, 1)

combinations()

combinations(iterable,r);创建一个迭代器,返回iterable中所有长度为r的子序列,返回的子序列中的项按输入iterable中的顺序排序

note:不带重复

for i in combinations([1, 2, 3], 2):
  print i
(1, 2)
(1, 3)
(2, 3)
combinations_with_replacement()

同上, 带重复 例子:

for i in combinations_with_replacement([1, 2, 3], 2):
  print i
(1, 1)
(1, 2)
(1, 3)
(2, 2)
(2, 3)
(3, 3)

应用示例

求质数序列中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)

总结

以上所述是小编给大家介绍的python itertools功能,希望对大家有所帮助!

Python 相关文章推荐
简单介绍Python中利用生成器实现的并发编程
May 04 Python
从django的中间件直接返回请求的方法
May 30 Python
使用pandas read_table读取csv文件的方法
Jul 04 Python
python 使用 requests 模块发送http请求 的方法
Dec 09 Python
python pygame实现五子棋小游戏
Oct 26 Python
Python使用tkinter模块实现推箱子游戏
Oct 08 Python
详解centos7+django+python3+mysql+阿里云部署项目全流程
Nov 15 Python
用python画一只可爱的皮卡丘实例
Nov 21 Python
Python模块 _winreg操作注册表
Feb 05 Python
python在不同条件下的输入与输出
Feb 13 Python
DRF框架API版本管理实现方法解析
Aug 21 Python
详解python polyscope库的安装和例程
Nov 13 Python
Python中itertools的用法详解
Feb 07 #Python
Python转换itertools.chain对象为数组的方法
Feb 07 #Python
已安装tensorflow-gpu,但keras无法使用GPU加速的解决
Feb 07 #Python
python十进制转二进制的详解
Feb 07 #Python
基于Tensorflow使用CPU而不用GPU问题的解决
Feb 07 #Python
python实现ip地址的包含关系判断
Feb 07 #Python
tensorflow:指定gpu 限制使用量百分比,设置最小使用量的实现
Feb 06 #Python
You might like
十天学会php之第四天
2006/10/09 PHP
如何用php获取文件名后缀
2013/06/09 PHP
thinkPHP框架整合tcpdf插件操作示例
2018/08/07 PHP
thinkphp5 框架结合plupload实现图片批量上传功能示例
2020/04/04 PHP
Javascript 验证上传图片大小[客户端]
2009/08/01 Javascript
超简单的jquery的AJAX用法
2010/05/10 Javascript
前后台交互过程中json格式如何解析以及如何生成
2012/12/26 Javascript
JavaScript eval() 函数介绍及应用示例
2014/07/29 Javascript
JQuery的Ajax中Post方法传递中文出现乱码的解决方法
2014/10/21 Javascript
jquery实现很酷的网页顶部图标下拉菜单效果
2015/08/22 Javascript
jquery实现仿新浪微博带动画效果弹出层代码(可关闭、可拖动)
2015/10/12 Javascript
js选择器全面解析
2016/06/27 Javascript
动态JavaScript所造成一些你不知道的危害
2016/09/25 Javascript
vue组件如何被其他项目引用
2017/04/13 Javascript
js读取本地文件的实例
2017/12/22 Javascript
vue的全局提示框组件实例代码
2018/02/26 Javascript
vue2.0学习之axios的封装与vuex介绍
2018/05/28 Javascript
利用layer实现表单完美验证的方法
2019/09/26 Javascript
[02:25]DOTA2英雄基础教程 生死判决瘟疫法师
2013/12/06 DOTA
Python序列之list和tuple常用方法以及注意事项
2015/01/09 Python
Python编程判断一个正整数是否为素数的方法
2017/04/14 Python
pyqt5使用按钮进行界面的跳转方法
2019/06/19 Python
python pandas生成时间列表
2019/06/29 Python
python实现抠图给证件照换背景源码
2019/08/20 Python
Python+OpenCv制作证件图片生成器的操作方法
2019/08/21 Python
Python 装饰器原理、定义与用法详解
2019/12/07 Python
基于Python爬虫采集天气网实时信息
2020/06/05 Python
HTML5的结构和语义(5):内嵌媒体
2008/10/17 HTML / CSS
世界闻名的衬衫制造商:Savile Row Company
2018/07/30 全球购物
Under Armour瑞典官方网站:美国高端运动科技品牌
2018/11/21 全球购物
电子商务专业学生职业生涯规划
2014/03/07 职场文书
元旦联欢会策划方案
2014/06/11 职场文书
毕业实习证明范本
2015/06/16 职场文书
六一文艺汇演主持词
2015/06/30 职场文书
2016年政治理论学习心得体会
2016/01/25 职场文书
导游词之南京莫愁湖公园
2019/11/13 职场文书