Revista Informática

Filter, map y reduce en un diccionario de Python

Publicado el 20 enero 2020 por Daniel Rodríguez @analyticslane

En una entrada anterior realicé una introducción a la programación funcional en Python. Una entrada donde se explicaba cómo utilizar filter(), map() y reduce() sobre listas o tuplas. Pero no sobre diccionarios. En esta entrada se va a explicar cómo hacer utilizar filter, map o reduce en un diccionario de Python.

Diccionarios frente listas o tuplas en programación funcional

Los diccionarios se diferencian de las listas o tuplas en que en estos objetos se asocia una clave a un valor. No es una sucesión ordenada de valores. Lo que es necesario tener en cuenta a la hora de definir las funciones lambda a utilizar en los métodos filter(), map() o reduce(). Ahora en cada uno de los elementos existen dos valores: la clave y el valor. Mediante el método items() se puede extraer cada uno de estos en una tupla de dos elementos. Siendo este los valores que se han de pasar al método que se desee emplear.

filter() en diccionarios

Para filtrar los valores de un diccionario se puede utilizar el método filter(). Como es un diccionario se tiene en lugar de la variable se tiene que pasar los datos que se obtienen mediante el método items(). En este caso la función lambda que se cree recibirá en cada uno de cados una tupla con dos elementos. Siendo el primero la clave y el segundo el valor. Así para filtrar los elementos por valor se puede usar algo como el siguiente donde se seleccionan los valores mayores de uno :

variables = {'A':1, 'B':2, 'C':3}

dict(filter(lambda x: x[1] > 1, variables.items()))
{'B': 2, 'C': 3}

Siendo necesario convertir la salida del método filter() en un diccionario.

En este caso también se puede definir una función lambda para filtrar en base a la clave y no el valor. Para eso solamente se tienen que utilizar el primer elemento de la tupla. Esto es lo que se puede ver en el siguiente caso donde se seleccionar el elemento con la clave B.

dict(filter(lambda x: x[0] == 'B', variables.items()))
{'B': 2}

Aunque este ejemplo es trivial la función lambda puede ser todo lo compleja que se desee.

map() en diccionarios

El método map() también se puede utilizar del mismo modo con diccionarios. Por ejemplo, multiplicar por un valor todos los valores de un diccionario.

dict(map(lambda x: (x[0], x[1] * 2), variables.items()))
{'A': 2, 'B': 4, 'C': 6}

reduce() en diccionarios

Finalmente, para el método reduce () es necesario tener en cuenta que el resultado del método será el primer valor de la función lambda en la segunda iteración. Por lo que este ha de ser compatible con la operación. Si se quiere obtener un valor numérico, por ejemplo, la suma de los elementos la mejor opción es dar un valor inicial al método. Esto es lo que se muestra en el siguiente ejemplo.

from functools import reduce

reduce(lambda x, y: x + y[1], variables.items(), 0)
6

Si se omite el valor inicial, el código fallará porque x en la primera iteración será una tupla y no un valor numérico.

Conclusiones

En esta entrada se ha visto cómo emplear filter, map o reduce en un diccionario de Python. Complementado así la entrad en la que se habló de programación funcional en Python. Básicamente hay que tener en cuenta que, a diferencia de las listas o tuplas, en los diccionarios hay dos valores la clave y el valor. Lo que obliga a definir las funciones lambda en función a esto.

Imágenes: Pixabay (PDPics)


Volver a la Portada de Logo Paperblog