Apuntes_Python/02_conceptos/07_itertools/README.md

205 lines
4.7 KiB
Markdown
Raw Normal View History

2022-12-24 22:41:20 -03:00
# 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](https://gitea.kickto.net/devfzn/Apuntes_Python/src/branch/master/02_conceptos/07_itertools#product)
- [permutations](https://gitea.kickto.net/devfzn/Apuntes_Python/src/branch/master/02_conceptos/07_itertools#permutations)
- [combinations, combinations_with_replacement](https://gitea.kickto.net/devfzn/Apuntes_Python/src/branch/master/02_conceptos/07_itertools#combinations)
- [accumulate (multiplicación, max)](https://gitea.kickto.net/devfzn/Apuntes_Python/src/branch/master/02_conceptos/07_itertools#accumulate)
- [groupby](https://gitea.kickto.net/devfzn/Apuntes_Python/src/branch/master/02_conceptos/07_itertools#groupby)
- [infinite iterators (count, cycle, repeat)](https://gitea.kickto.net/devfzn/Apuntes_Python/src/branch/master/02_conceptos/07_itertools#infinite-iterators)
### product
```python
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
```python
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).
```python
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
```python
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).
```python
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
```python
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
```python
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
```python
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
```python
from itertools import repeat
for i in repeat(1, 4):
print(i)
# 1
# 1
# 1
# 1
```