4.7 KiB
itertools
El módulo itertools es un conjunto de herramientas para manejar iteradores.
Los iteradores son un tipo de dato que usar para iterar, ejm. en un loop for.
Un iterator es un objeto sobre el cual se puede iterar, es decir, se pueden
recorrer todos sus valores.
Un iterador es un objeto que contiene un número contable de valores.
En python un iterador es un obj. que implementa el protocolo iterator, que
consiste en los métodos __iter__() y __next__().
Listas, tuplas, diccionarios y sets, son objetos iterables. De los cuales se
puede obtener un iterador, metodo iter()
Algunas herramientas de los iteradores son:
- product
- permutations
- combinations, combinations_with_replacement
- accumulate (multiplicación, max)
- groupby
- infinite iterators (count, cycle, repeat)
product
from itertools import product
a = [1, 2]
b = [3, 4]
prod = product(a, b) # <itertools.product object at ...>
list(prod)
# [(1, 3), (1, 4),
# (2, 3), (2, 4)]
a = [1, 2]
b = [3]
prod = product(a, b, repeat=2) # <itertools.product object at ...>
list(prod)
# [(1, 3, 1, 3), (1, 3, 2, 3),
# (2, 3, 1, 3), (2, 3, 2, 3)]
permutations
Todos los posibles ordenes de una entrada
from itertools import permutations
a = [1,2,3]
perm = permutations(a)
list(perm)
# [(1, 2, 3), (1, 3, 2),
# (2, 1, 3), (2, 3, 1),
# (3, 1, 2), (3, 2, 1)]
# especificar largo de permutaciones
perm = permutations(a, 2)
list(perm)
# [(1, 2), (1, 3),
# (2, 1), (2, 3),
# (3, 1), (3, 2)]
combinations
Realiza todas las posibles combinaciones, según largo
pasado como 2do argumento (no repite elementos).
from itertools import (combinations,
combinations_with_replacement,
accumulate)
a = [1,2,3,4]
comb = combinations(a, 2)
list(comb)
# [(1, 2), (1, 3),
# (1, 4), (2, 3),
# (2, 4), (3, 4)]
combinations_with_replacement
Igual que el anterior pero repite elementos
comb_wr = combinations_with_replacement(a, 2)
list(comb_wr)
# [(1, 1), (1, 2), (1, 3), (1, 4), (2, 2),
# (2, 3), (2, 4), (3, 3), (3, 4), (4, 4)]
accumulate
Acumula(suma o concatena) elemementos (del mismo tipo).
accum = accumulate(a)
list(accum) # [1,3,6,10]
b = ['a', 'b', 'cedario']
accum2 = accumulate(b)
list(accum2) # ['a', 'ab', 'abcedario']
# accumulate, multiplicación
import operator
accum_mult = accumulate(a, func=operator.mul)
list(accum_mult) # [1, 2, 6, 24]
# accumulate, max
c = [1, 2, 5, 3, 4]
accm_max = accumulate(c, func=max)
list(accm_max) # [1, 2, 5, 5, 5]
groupby
Crea un iterador que retorna keys y groups de un elemento iterable
from itertools import groupby
#def menor_q_3(x):
# return x < 3
# reemplazado por funcion lambda
a = [1, 2, 3, 4]
group_obj = groupby(a, key=lambda x: x<3)
for key, value in group_obj:
print(key, list(value))
# True [1, 2]
# False [3, 4]
personas = [{'nombre': 'Zerafio', 'edad': 26},
{'nombre': 'Raber', 'edad': 26},
{'nombre': 'Estor', 'edad': 58},
{'nombre': 'Gunter', 'edad': 26}]
group_obj = groupby(personas, key=lambda x: x['edad'])
for key, value in group_obj:
print(key, list(value))
# agrupa solo si los elementos son consecutivos?
# 26 [{'nombre': 'Zerafio', 'edad': 26}, {'nombre': 'Raber', 'edad': 26}]
# 58 [{'nombre': 'Estor', 'edad': 58}]
# 26 [{'nombre': 'Gunter', 'edad': 26}]
infite iterators
count
Itera infinitamente desde un numero
from itertools import count
for i in count(10):
print(i)
if i == 15:
break
# 10
# 11
# 12
# 13
# 14
# 15
cycle
Itera infinitamente un iterable
from itertools import cycle
a = [1, 2, 3]
cont = 0
for i in cycle(a):
print(i)
cont += 1
if cont > 8:
break
# 1
# 2
# 3
# 1
# 2
# 3
repeat
Repite el argumento pasado infinitamente,
o hasta el valor pasado como segundo argumento
from itertools import repeat
for i in repeat(1, 4):
print(i)
# 1
# 1
# 1
# 1