python的itertools包怎么使用

這篇文章主要介紹了python的itertools包怎么使用的相關(guān)知識(shí)，內(nèi)容詳細(xì)易懂，操作簡(jiǎn)單快捷，具有一定借鑒價(jià)值，相信大家閱讀完這篇python的itertools包怎么使用文章都會(huì)有所收獲，下面我們一起來看看吧。

前言：

Python的內(nèi)建模塊itertools提供了非常有用的用于操作迭代對(duì)象的函數(shù)，itertools提供的幾個(gè)“無限”迭代器

1、count()創(chuàng)建自然數(shù)序列

import itertools
 
n = 0
natuals = itertools.count(1)
for m in natuals:
    print(m)

   因?yàn)?code>count(n)會(huì)創(chuàng)建一個(gè)無限的迭代器，起始數(shù)字為n，迭代至無窮大。所以上述代碼會(huì)打印出自然數(shù)序列，根本停不下來，只能按Ctrl+C退出。

運(yùn)行結(jié)果：

0 1 2 3 4 5 6 7 ...

2、cycle()創(chuàng)建序列循環(huán)

        cycle()會(huì)把傳入的一個(gè)序列無限重復(fù)下去。

示例1、循環(huán)列表中的內(nèi)容：

natuals = itertools.cycle([1, 2, 3])
for m in natuals:
    print(m)

運(yùn)行結(jié)果：

1 2 3 1 2 3 1 2 3 ...

示例2、字符串中字符循環(huán)：

import itertools
cs = itertools.cycle('ABC') # 注意字符串也是序列的一種
for c in cs:
    print(c)

運(yùn)行結(jié)果：

A B C A B C A B C ...

3、repeat()把一個(gè)元素有限次重復(fù)

       repeat()負(fù)責(zé)把一個(gè)元素?zé)o限重復(fù)下去（默認(rèn)行為），不過如果提供第二個(gè)參數(shù)就可以限定重復(fù)次數(shù)。

ns = itertools.repeat('A', 3)
for n in ns:
    print(n)

運(yùn)行結(jié)果：

A A A ...

4、takewhile()根據(jù)條件判斷來截取出一個(gè)有限的序列

        無限序列只有在for迭代時(shí)才會(huì)無限地迭代下去，如果只是創(chuàng)建了一個(gè)迭代對(duì)象，它不會(huì)事先把無限個(gè)元素生成出來，事實(shí)上也不可能在內(nèi)存中創(chuàng)建無限多個(gè)元素。無限序列雖然可以無限迭代下去，但是通常我們會(huì)通過takewhile()等函數(shù)根據(jù)條件判斷來截取出一個(gè)有限的序列。

import itertools
 
natuals = itertools.count(1)
ns = itertools.takewhile(lambda x: x <= 10, natuals)
print(list(ns))

運(yùn)行結(jié)果：

[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

迭代器測(cè)試：

        對(duì)于有限數(shù)據(jù)的迭代器，每調(diào)用一次，迭代器內(nèi)數(shù)據(jù)彈出。

import itertools
 
natuals = itertools.count(1)
ns = itertools.takewhile(lambda x: x <= 3, natuals)
for n in ns:
    print(n)
 
print(list(ns))

運(yùn)行結(jié)果：

1 2 3 []

5、chain()迭代對(duì)象串聯(lián)

        chain()可以把一組迭代對(duì)象串聯(lián)起來，形成一個(gè)更大的迭代器。

for c in itertools.chain('ABC', 'XYZ'):
    print(c)

運(yùn)行結(jié)果：

'A' 'B' 'C' 'X' 'Y' 'Z'

6、groupby()迭代器元素分組

       groupby()把迭代器中相鄰的重復(fù)元素挑出來放在一起，更多的用于字符串。

for key, group in itertools.groupby('AAABBBCCAAA'):
    print(key, list(group))

運(yùn)行結(jié)果：

A ['A', 'A', 'A']
B ['B', 'B', 'B']
C ['C', 'C']
A ['A', 'A', 'A']

        實(shí)際上挑選規(guī)則是通過函數(shù)完成的，只要作用于函數(shù)的兩個(gè)元素返回的值相等，這兩個(gè)元素就被認(rèn)為是在一組的，而函數(shù)返回值作為組的key。如果我們要忽略大小寫分組，就可以讓元素'A'和'a'都返回相同的key。

for key, group in itertools.groupby('AaaBBbcCAAa', lambda c: c.upper()):
    print(key, list(group))

運(yùn)行結(jié)果：

A ['A', 'a', 'a']
B ['B', 'B', 'b']
C ['c', 'C']
A ['A', 'A', 'a']

關(guān)于“python的itertools包怎么使用”這篇文章的內(nèi)容就介紹到這里，感謝各位的閱讀！相信大家對(duì)“python的itertools包怎么使用”知識(shí)都有一定的了解，大家如果還想學(xué)習(xí)更多知識(shí)，歡迎關(guān)注億速云行業(yè)資訊頻道。