itérateur python et module itertools

Itérateur En python, le protocole itérateur consiste à implémenter la méthode __iter() et la méthode next() de l'objet, où la première renvoie l'objet lui-même, qui renvoie le élément suivant du conteneur. Les objets qui implémentent ces deux méthodes sont des objets itérables. Les itérateurs sont paresseux et ne sont générés que lorsqu'ils sont utilisés, ce qui offre des avantages pour le traitement de grandes quantités de données, contrairement à l'écriture simultanée de toutes les données en mémoire. Ci-dessous, j'ai moi-même écrit un itérateur. Vous pouvez voir que vous pouvez utiliser une boucle for pour traiter l'itérateur que vous avez écrit. Pour les objets qui implémentent le protocole itérateur, vous pouvez utiliser n'importe quel outil itérateur similaire à une boucle for. Cependant, en regardant le résultat ci-dessous, le deuxième résultat est vide. Pourquoi ? Lorsque nous utilisons une liste, nous pouvons afficher le même objet plusieurs fois. Quelle est la différence entre celui-ci et un objet qui implémente son propre protocole d'itérateur ?

class it(object):

def __init__(self, n):

self.a = 0
self.n = n

def __iter__(self):
return self

<br/>

def next(self):
if self.a < self.n:
self.a = 1
return self.a
else :
Augmenter la startration

i = it (5)
pour j dans i :
Print j,
PRINT '
PRINT' ------
pour j in i:
print j
# 1 2 3 4 5
# ------
Après avoir étudié, j'ai appris que la liste et d'autres types d'itérateurs renvoient un objet itérateur et ne retournent pas lui-même. Ensuite, j'ai écrit le code suivant à des fins de test. Comme vous pouvez le voir sur l'impression, des objets comme TestIt peuvent être utilisés à plusieurs reprises. Il y a donc une autre question. Un objet qui n’implémente pas la méthode next() est-il toujours un objet itérateur ? En effet, lors de l'utilisation de la classe it, un objet itérateur est renvoyé et la fonction itérateur est implémentée à l'aide de l'itérateur it, ce qui équivaut à implémenter le protocole itérateur. Le protocole itérateur est très utile en python. Il existe un module itertools en python sur les itérateurs. Je vais maintenant découvrir le module itertools et voir quelles surprises il y a !

class TestIt(object):
def __init__(self, a):
self.a = a 
def __iter__(self):
return it(self.a)

itertools

Itérateur infini

1 count(), accepte deux paramètres, Le premier est le numéro de départ, le second est la foulée, à partir de 0 par défaut, l'utilisation est la suivante

import itertools as it

c = it.count(10, 2)
for i in c:
    if i > 20:
        break
    print i,
# 10 12 14 16 18 20

2 cycle(), accepte un paramètre, qui est l'itération L'objet itérateur (liste, chaîne, etc.) bouclera pour générer les éléments dans l'itérateur

c = it.cycle([1, 2, 3])
i = 1
for j in c:
    if i > 7:
        break
    print j,
    i += 1

3 repeat(), qui accepte deux paramètres et est utilisé pour générer le premier paramètre n fois

for j in it.repeat([1, 2, 3], 4):
print j

itérateur limité, choisissez celui que vous pensez utiliser couramment et introduisez

1 chain(), qui accepte plusieurs itérateurs Objets en paramètres et connectez-les en chaîne('abc', [1, 2, 3])

2 compresser(données, sélecteurs), filtrer les précédents en fonction de ces derniers paramètres Paramètres, les deux paramètres doivent être des objets itérateurs

3

dropwhile(pre, iterable ) , le paramètre pre est une fonction Lorsque pre(i) est True, renvoie cet élément et tous les éléments suivants .

4 groupby(itérable[, keyfunc]), où iterable est un objet itérable, keyfunc est une fonction de regroupement, utilisée pour regrouper des éléments consécutifs dans un itérable. Si elle n'est pas spécifiée, elle regroupera par défaut des éléments identiques consécutifs dans un itérable et renverra un itérateur de . (key, sub-iterator)

5 ifilter(function or None, sequence)，将 iterable 中 function(item) 为 True 的元素组成一个迭代器返回，如果 function 是 None，则返回 iterable 中所有计算为 True 的项

6 tee(iterable [,n])，tee 用于从 iterable 创建 n 个独立的迭代器，以元组的形式返回，n 的默认值是 2。

 for j in it.tee('abc', 4):
     print list(j)

组合生成器

1 permutations(iterable[, r])，用于生成一个排列，r是生成排列的元素长度，不指定则为默认长度

  list(it.permutations(  list(it.permutations(, 2

2 combinations(iterable, r)， 求序列的组合，其中，r 指定生成组合的元素的长度,是必需的参数

3 combinations_with_replacement(iterable, r)，生成的组合包含自身元素 

 print list(it.combinations_with_replacement('abc', 2))
 # [('a', 'a'), ('a', 'b'), ('a', 'c'), ('b', 'b'), ('b', 'c'), ('c', 'c')]

更多python迭代器以及itertools模块相关文章请关注PHP中文网！