脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → python → Python 循环器itertools介绍

Python标准库之循环器(itertools)介绍

 更新时间：2014年11月25日 09:35:34   投稿：junjie  
这篇文章主要介绍了Python标准库之循环器(itertools)介绍,本文讲解了无穷循环器、函数式工具、组合工具、groupby()、其它工具等内容,需要的朋友可以参考下

在循环对象和函数对象中，我们了解了循环器(iterator)的功能。循环器是对象的容器，包含有多个对象。通过调用循环器的next()方法 (__next__()方法，在Python 3.x中)，循环器将依次返回一个对象。直到所有的对象遍历穷尽，循环器将举出StopIteration错误。

在for i in iterator结构中，循环器每次返回的对象将赋予给i，直到循环结束。使用iter()内置函数，我们可以将诸如表、字典等容器变为循环器。比如:

复制代码 代码如下:

for i in iter([2, 4, 5, 6]):
    print(i)

标准库中的itertools包提供了更加灵活的生成循环器的工具。这些工具的输入大都是已有的循环器。另一方面，这些工具完全可以自行使用Python实现，该包只是提供了一种比较标准、高效的实现方式。这也符合Python“只有且最好只有解决方案”的理念。

复制代码 代码如下:

# import the tools
from itertools import *

无穷循环器

count(5, 2)     #从5开始的整数循环器，每次增加2，即5, 7, 9, 11, 13, 15 ...
cycle('abc')    #重复序列的元素，既a, b, c, a, b, c ...
repeat(1.2)     #重复1.2，构成无穷循环器，即1.2, 1.2, 1.2, ...

repeat也可以有一个次数限制:

repeat(10, 5)   #重复10，共重复5次

函数式工具

函数式编程是将函数本身作为处理对象的编程范式。在Python中，函数也是对象，因此可以轻松的进行一些函数式的处理，比如map(), filter(), reduce()函数。

itertools包含类似的工具。这些函数接收函数作为参数，并将结果返回为一个循环器。

比如：

复制代码 代码如下:

from itertools import *
rlt = imap(pow, [1, 2, 3], [1, 2, 3])
for num in rlt:
    print(num)

上面显示了imap函数。该函数与map()函数功能相似，只不过返回的不是序列，而是一个循环器。包含元素1, 4, 27，即1**1, 2**2, 3**3的结果。函数pow(内置的乘方函数)作为第一个参数。pow()依次作用于后面两个列表的每个元素，并收集函数结果，组成返回的循环器。

此外，还可以用下面的函数:

复制代码 代码如下:

starmap(pow, [(1, 1), (2, 2), (3, 3)])

pow将依次作用于表的每个tuple。

ifilter函数与filter()函数类似，只是返回的是一个循环器。

复制代码 代码如下:

ifilter(lambda x: x > 5, [2, 3, 5, 6, 7]

将lambda函数依次作用于每个元素，如果函数返回True，则收集原来的元素。6, 7

此外，

复制代码 代码如下:

ifilterfalse(lambda x: x > 5, [2, 3, 5, 6, 7])

与上面类似，但收集返回False的元素。2, 3, 5
复制代码 代码如下:

takewhile(lambda x: x < 5, [1, 3, 6, 7, 1])

当函数返回True时，收集元素到循环器。一旦函数返回False，则停止。1, 3
复制代码 代码如下:

dropwhile(lambda x: x < 5, [1, 3, 6, 7, 1])

当函数返回False时，跳过元素。一旦函数返回True，则开始收集剩下的所有元素到循环器。6, 7, 1。

组合工具

我们可以通过组合原有循环器，来获得新的循环器。

复制代码 代码如下:

chain([1, 2, 3], [4, 5, 7])      # 连接两个循环器成为一个。1, 2, 3, 4, 5, 7
product('abc', [1, 2])   # 多个循环器集合的笛卡尔积。相当于嵌套循环 
for m, n in product('abc', [1, 2]):
    print m, n

permutations('abc', 2)   # 从'abcd'中挑选两个元素，比如ab, bc, ... 将所有结果排序，返回为新的循环器。

注意，上面的组合分顺序，即ab, ba都返回。

combinations('abc', 2)   # 从'abcd'中挑选两个元素，比如ab, bc, ... 将所有结果排序，返回为新的循环器。

注意，上面的组合不分顺序，即ab, ba的话，只返回一个ab。

combinations_with_replacement('abc', 2) # 与上面类似，但允许两次选出的元素重复。即多了aa, bb, cc。

groupby()

将key函数作用于原循环器的各个元素。根据key函数结果，将拥有相同函数结果的元素分到一个新的循环器。每个新的循环器以函数返回结果为标签。

这就好像一群人的身高作为循环器。我们可以使用这样一个key函数: 如果身高大于180，返回"tall"；如果身高底于160，返回"short";中间的返回"middle"。最终，所有身高将分为三个循环器，即"tall", "short", "middle"。

复制代码 代码如下:

def height_class(h):
     if h > 180:
        return "tall"
     elif h < 160:
        return "short"
     else:
        return "middle"friends = [191, 158, 159, 165, 170, 177, 181, 182, 190] friends = sorted(friends, key = height_class)
 
for m, n in groupby(friends, key = height_class):
    print(m)
    print(list(n))

注意，groupby的功能类似于UNIX中的uniq命令。分组之前需要使用sorted()对原循环器的元素，根据key函数进行排序，让同组元素先在位置上靠拢。

其它工具

compress('ABCD', [1, 1, 1, 0])  # 根据[1, 1, 1, 0]的真假值情况，选择第一个参数'ABCD'中的元素。A, B, C
islice()                        # 类似于slice()函数，只是返回的是一个循环器
izip()                          # 类似于zip()函数，只是返回的是一个循环器。

总结

itertools的工具都可以自行实现。itertools只是提供了更加成形的解决方案。

您可能感兴趣的文章:

相关文章

  • 什么是python的列表推导式

    什么是python的列表推导式

    在本篇文章里小编给大家分享了关于python列表推导式的含义及用法，需要的朋友们可以参考下。
    2020-05-05
  • Python遍历zip文件输出名称时出现乱码问题的解决方法

    Python遍历zip文件输出名称时出现乱码问题的解决方法

    这篇文章主要介绍了Python遍历zip文件输出名称时出现乱码问题的解决方法,实例分析了Python乱码的出现的原因与相应的解决方法,需要的朋友可以参考下
    2015-04-04
  • Python中使用ctypes调用C++的方法

    Python中使用ctypes调用C++的方法

    本文主要介绍了Python中使用ctypes调用C++的方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2023-06-06
  • Python Django教程之模板的使用

    Python Django教程之模板的使用

    模板是 Django MVT 结构的第三个也是最重要的部分。Django中的模板基本上是用HTML，CSS和Javascript编写在.html文件中的。本文将通过实例详细聊聊Django模板的使用，感兴趣的可以
    2022-10-10
  • Python容器使用的5个技巧和2个误区总结

    Python容器使用的5个技巧和2个误区总结

    在本篇文章里小编给大家整理的是关于Python容器使用的5个技巧和2个误区的相关知识点内容，需要的朋友们学习下。
    2019-09-09
  • python七种方法判断字符串是否包含子串

    python七种方法判断字符串是否包含子串

    这篇文章主要介绍了python七种方法判断字符串是否包含子串，帮助大家更好的理解和学习python，感兴趣的朋友可以了解下
    2020-08-08
  • TensorFlow2.4完成Word2vec词嵌入训练方法详解

    TensorFlow2.4完成Word2vec词嵌入训练方法详解

    这篇文章主要为大家介绍了TensorFlow2.4完成Word2vec词嵌入训练方法详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
    2022-11-11
  • wxPython：python首选的GUI库实例分享

    wxPython：python首选的GUI库实例分享

    wxPython是Python语言的一套优秀的GUI图形库。允许Python程序员很方便的创建完整的、功能键全的GUI用户界面。 wxPython是作为优秀的跨平台GUI库wxWidgets的Python封装和Python模块的方式提供给用户的
    2019-10-10
  • python 列表常用方法超详细梳理总结

    python 列表常用方法超详细梳理总结

    这篇文章主要为大家介绍了Python中列表的几个常用方法总结，文中的示例代码讲解详细，对我们学习Python列表有一定帮助，需要的可以参考一下
    2022-03-03
  • Python内置函数hex()的实现示例

    Python内置函数hex()的实现示例

    这篇文章主要介绍了Python内置函数hex()的实现示例，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2020-08-08

最新评论