前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >itertools

itertools

作者头像
TomatoCool
发布2023-08-10 14:22:39
1630
发布2023-08-10 14:22:39
举报
文章被收录于专栏:TomatoCool

导入

代码语言:javascript
复制
from itertools import *

无限迭代器

count()

作用: 无限生成数字,可设置步长。

格式:

代码语言:javascript
复制
count(start, step=1)

使用示例:

代码语言:javascript
复制
#  打印0,-2,-4,-6...
for i in count(0, step=-2):
    print(i)

cycle()

作用: 无限生成字符。

格式:

代码语言:javascript
复制
cycle(string)

使用示例:

代码语言:javascript
复制
#  打印a,b,c,a,b...
for i in cycle("abc"):
    print(i)

repeat()

作用: 无限打印元素。

格式:

代码语言:javascript
复制
repeat(item, n)

使用示例:

代码语言:javascript
复制
#  打印a,a,a,a,a...
for i in repeat("a"):
    print(i)
#  打印b,b,b
for i in repeat("b", 3):
    print(i)

有限迭代器

chain()

作用: 将迭代器合并。

格式:

代码语言:javascript
复制
chain(a1, a2, a3, ....)

使用示例:

代码语言:javascript
复制
#  输出a,b,c,1,2,3
for i in chain("abc", "123"):
    print(i)

groupby()

作用: 把迭代器中相邻的元素按照key函数分组,当key=None时,把相邻的重复元素进行分组。

格式:

代码语言:javascript
复制
groupby(iterable, key=None)

使用示例:

代码语言:javascript
复制
#  对相邻的重复元素分组
for key group in chain("aaabbbcccdd"):
    print(key, group)
#  对大写相同的相邻元素分组
for key group in chain("AaaBbbcCcDd", lambda x: x.upper()):
    print(key, group)

accumulate()

作用: 计算迭代器,如果不指定参数函数,将会采取默认的求和函数。(实际上是对序列的顺序子序列使用func函数并返回结果,比如说max(iterable[:1]),max(iterable[:2]),max(iterable[:3])...)

格式:

代码语言:javascript
复制
accumulate(iterable, func=sum)

使用示例:

代码语言:javascript
复制
#  分别对[:0],[:1],[:2]求和
for i in accumulate([1, 2, 3, 4, 5, 6], func=sum):
    print(i)
#  求前i个元素的中最大值
for i in accumulate([1, 2, 3, 4, 5, 6], func=max):
    print(i)

组合迭代器

product()

作用: 笛卡尔积,相当于嵌套的for循环。

格式:

代码语言:javascript
复制
product(p, q, …, repeat=1)

使用示例:

代码语言:javascript
复制
#  输出ac,ad,bc,bd
for i in product("ab", "cd", …, repeat=1):
    print(i)
#  输出aa,ab,ba,bb
for i in product("ab"repeat=2):
    print(i)

permutations()

作用: 排列,可指定长度。

格式:

代码语言:javascript
复制
permutations(p,  r=len(p))

使用示例:

代码语言:javascript
复制
#  从序列中抽取两个元素的排列
for i in permutations("abcd", 2)
    print(i)

combinations()

作用: 组合,无重复元素,可指定长度。

格式:

代码语言:javascript
复制
combinations(p, r=len(p))

使用示例:

代码语言:javascript
复制
#  从序列中抽取两个元素的组合
for i in combinations("abcd", 2)
    print(i)

combinations_with_replacement()

作用: 组合,有重复元素,可指定长度。

格式:

代码语言:javascript
复制
combinations_with_replacement(p, r=len(p))

使用示例:

代码语言:javascript
复制
#  从序列中抽取两个元素的组合
for i in combinations_with_replacement("abcd", 2)
    print(i)
本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2023-08-08,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 导入
  • 无限迭代器
    • count()
      • cycle()
        • repeat()
        • 有限迭代器
          • chain()
            • groupby()
              • accumulate()
              • 组合迭代器
                • product()
                  • permutations()
                    • combinations()
                      • combinations_with_replacement()
                      领券
                      问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档