Ordenar listas de objetos en Python

Publicado el 16 marzo 2020 por Daniel Rodríguez @analyticslane

Cuando queremos ordenar una lista en Python se puede utilizar el método sorted(). Un método que nos devuelve una lista con los objetos ordenados. Aunque si el contenido de la lista no es no son datos primitivos, como por ejemplo tuplas, diccionarios u objetos el método no funcionará. Para ello es necesario pasar una función con la que sorted() pueda ordenar los datos. Siendo esto lo que veremos a continuación: cómo ordenar listas de objetos en Python.

Ordenar una lista de tipos primitivos

Si queremos ordenar una lista de números o caracteres solo hay que pasar la lista al método sorted() para que este nos devuelva una nueva lista ordenada. Como el ejemplo que se muestra a continuación.

lista = [2 ,7, 2, 1, 6, 3]

sorted(lista)
[1, 2, 2, 3, 6, 7]

El método tiene la opción de cambiar la forma de ordenación, por defecto es ascendente, pero se puede indicar que sea descendente asignando el valor verdadero a la propiedad reverse. Lo que se puede ver en el siguiente ejemplo

sorted(lista, reverse=True)
[7, 6, 3, 2, 2, 1]

Ordenación de tuplas

En el caso de que tengamos una lista de tuplas, en las que cada valor significa algo, el método sorted() ordena la lista en base al primer valor. Como se muestra a continuación:

tuplas = [(1, 14), (2, 12), (0, 22), (3, 13)]

sorted(tuplas)
[(0, 22), (1, 14), (2, 12), (3, 13)]

Si esto es lo que se desea no hay ningún problema, pero no es extraño que necesitemos ordenar la tupla por el segundo, tercero o incluso la suma de los elementos. Para esto se le puede indicar al método un criterio de ordenación mediante la propiedad key. Para lo que generalmente se utilizará una función lambda. Por ejemplo, para ordenar por el segundo elemento:

sorted(tuplas, key=lambda x: x[1])
[(2, 12), (3, 13), (1, 14), (0, 22)]

Otra opción puede ser usar la suma de las tuplas para ordenar esta.

sorted(tuplas, key=lambda x: sum(x))
[(2, 12), (1, 14), (3, 13), (0, 22)]

Ordenación de diccionarios

El caso de los diccionarios es similar al de las tuplas, solo que ahora el método no tiene un método de ordenación por defecto, por lo que es necesario indicarlo siempre. Ya que si intentamos ordenar un diccionario sin esta función el método sorted() nos devolverá un error. Así un diccionario con la propiedad edad se puede ordenar fácilmente como se muestra en el siguiente ejemplo.

dictionary = [{'edad':23}, {'edad':21}, {'edad':44}, {'edad':19}]

sorted(dictionary, key=lambda x: x['edad'])
[{'edad': 19}, {'edad': 21}, {'edad': 23}, {'edad': 44}]

Ordenación de objetos

Finalmente, si lo que deseamos es ordenar una lista de objetos dependerá de cómo se hubiesen definidos estos. En el caso de que se hubiese definido un método __gt__ en la clase sorted() funcionara sin problema, teniendo en cuenta este valor. En caso contrario es necesario indicar la función de ordenación. Por ejemplo, si se crea una clase Persona con las propiedades nombre y edad y no se implementa el método __gt__ una lista de estos objetos se tiene que ordenar indicando a sorted() el criterio. Ya que en caso contrario sorted() daría un error.

class Persona():
    def __init__(self, nombre, edad):
        self.nombre = nombre
        self.edad = edad
        
    def __repr__(self):
        return "{0}:{1}".format(self.nombre, self.edad) 
        
personas = [Persona('Juan', 23), Persona('Pedro', 18), Persona('Antonio', 33)]

sorted(personas, key=lambda x: x.edad)
[Pedro:18, Juan:23, Antonio:33]

Por otro lado, si se define el método __gt__ no es necesario indicar el criterio de ordenación.

class Persona():
    def __init__(self, nombre, edad):
        self.nombre = nombre
        self.edad = edad
    
    def __repr__(self):
        return "{0}:{1}".format(self.nombre, self.edad) 
    
    def __gt__(self, persona):
        return self.edad > persona.edad

    
personas = [Persona('Juan', 23), Persona('Pedro', 18), Persona('Antonio', 33)]

sorted(personas)
[Pedro:18, Juan:23, Antonio:33]

Salvo que se desee ordenar en base a un criterio diferente, como ordenar alfabeticamente por nombre el lugar de por edad.

sorted(personas, key=lambda x: x.nombre)
[Antonio:33, Juan:23, Pedro:18]

Conclusiones

En esta entrada se ha visto cómo ordenar listas de objetos en Python. Comprobado como se puede utilizar el método sorted() no solo para ordenar valores primitivos, sino que cualquier componente como puede ser una tupla, diccionario u objeto.

Imagen de fancycrave1 en Pixabay


Publicidad