Cómo usar el método pop() de Python

El método pop() es una función importante de Python que se utiliza en la manipulación de datos para eliminar y devolver un elemento específico de una lista o un diccionario. Quienes programan en Python suelen preferir pop() porque es una forma muy sencilla de modificar estructuras de datos como listas y diccionarios.

Antes de entrar en detalle, te recomiendo encarecidamente el curso Introduction to Python de DataCamp para aprender los fundamentos del lenguaje. La Python Cheat Sheet for Beginners también te ayudará a consultar y entender la sintaxis de Python y sus diferentes funciones.

Respuesta rápida: cómo usar el método pop() en Python

El método pop() se usa en listas y diccionarios para eliminar elementos específicos y devolver su valor. Voy a mostrar un ejemplo rápido de ambos casos. El primer ejemplo muestra cómo eliminar un elemento de una lista usando pop(). El siguiente código imprime 5.

# Create a list
my_list = [1, 2, 3, 4, 5]

# Removes and returns the last element of the list
last_element = my_list.pop()
print(last_element)

El siguiente ejemplo muestra cómo usar pop() para eliminar una clave concreta de un diccionario. El siguiente código imprime 3.

# Create a dictionary
my_dict = {'a': 1, 'b': 2, 'c': 3}

# Removes and returns the value associated with the key 'b'
value = my_dict.pop('c')
print(value)

Usar pop() con listas

El método pop() se usa con listas para eliminar y devolver el elemento indicado de esa lista. La sintaxis de pop() con listas es la siguiente.

list.pop([index])

Cuando usas pop() con listas, ten en cuenta que puede recibir un parámetro opcional para especificar el índice a eliminar. Si no se proporciona parámetro, pop() elimina y devuelve el último elemento de la lista.

Si, en cambio, decides indicar el índice explícitamente, ten presente que si lo haces de forma incorrecta se lanzará un IndexError si el índice especificado está fuera de rango. Veamos a continuación los diferentes casos de uso de pop() con listas.

Usar pop() para eliminar por índice

También puedes usar un índice positivo o negativo como argumento de pop() para indicar el elemento que quieres eliminar de la lista. En Python, los índices positivos empiezan en 0 al principio de la lista. Los índices negativos empiezan en -1 desde el final.

Los siguientes ejemplos muestran tres formas de eliminar elementos de una lista. En todos los casos, la salida imprimirá 5.

# Create a list
my_list = [1, 2, 3, 4, 5]

# Removes and returns the last element of the list
last_element = my_list.pop()
print(last_element)

# Create a list
my_list = [1, 2, 3, 4, 5]

# Remove specific element using negative index
removed_element = my_list.pop(-1) 
print(removed_element)  

# Create  a list
my_list = [1, 2, 3, 4, 5]

# Remove specific element using negative index
removed_element = my_list.pop(4) 
print(removed_element) 

Puedes usar pop() en un bucle para recorrer la lista y eliminar el último elemento cada vez hasta que quede vacía. El siguiente ejemplo muestra cómo usar pop() en bucles. El código imprimirá List is now empty.

# Create a list
my_list = [1, 2, 3, 4, 5]

# Create a function to remove items from the list
while my_list:
    element = my_list.pop()
    print(f"Removed element: {element}")
print("List is now empty:", my_list)

Esta técnica puede ser útil en varios escenarios. Imagina, por ejemplo, que tienes una lista de tareas para ir resolviéndolas una a una. Con pop() puedes quitar cada tarea cuando la completes y mantener la lista centrada en lo que queda por hacer. 

Usar pop() con listas anidadas

Puedes usar pop() para eliminar elementos de listas anidadas. Este método eliminará y devolverá la lista indicada dentro de la lista anidada. El siguiente código imprimirá [7, 8, 9].

# Define a nested list
nested_list = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

# Pop an element from the outer list
removed_list = nested_list.pop()
print(removed_list)

Si quieres eliminar un elemento concreto de una lista interna, primero indica el índice de esa lista anidada. El código de abajo imprime 9, que es el último elemento de la lista en el índice 2.

# Define a nested list
nested_list = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

# Pop an element from an inner list
removed_element = nested_list[2].pop()
print(removed_element)

Cómo gestionar el error "pop() index out of range"

Si intentas eliminar un elemento de una lista con un índice fuera del rango de esa lista, se lanzará un IndexError. Para evitarlo, es importante comprobar primero la longitud de la lista y luego proporcionar un índice válido. También puedes usar un bloque try-except para capturar el error. En ambos casos, el código imprimirá Index out of range.

# Create a list
my_list = [1, 2, 3]

# Check length before popping
index = 4

# Ensure the index is within the valid range of the list
if 0 <= index < len(my_list):
    # Removes and returns the element at the specified index
    element = my_list.pop(index)  
    print(f"Popped element: {element}")
else:
    print("Index out of range")

# Create a list
my_list = [1, 2, 3]

# Using try-except block to handle out-of-range index
try:
    # Attempt to remove and return the element at index 5
    element = my_list.pop(5)
except IndexError:
    # Catches the IndexError if the index is out of range
    print("Index out of range")

Usar pop() con diccionarios

El método pop() se usa con diccionarios para eliminar una clave específica y devolver su valor. La sintaxis de pop() con diccionarios es la siguiente.

dict.pop(key[, default])

Vemos que, al usar pop() con un diccionario, se requiere una clave como parámetro. El método pop() devuelve el valor asociado a la clave indicada y elimina ese par clave-valor del diccionario. 

También puedes proporcionar un valor por defecto para devolver si la clave no existe. Si la clave indicada no se encuentra y no se proporciona un valor por defecto, se lanza un KeyError. Si se da un valor por defecto, devuelve ese valor en su lugar.

Valor por defecto en diccionarios de Python

Al usar pop() con diccionarios, puedes indicar un valor por defecto que se devolverá si no se encuentra la clave. Esta técnica se utiliza normalmente para evitar el KeyError. El código de abajo imprime Not Found porque la clave d no está en la colección.

# Create a dictionary
my_dict = {'a': 1, 'b': 2, 'c': 3}

# Remove a specific key and provide default value
value = my_dict.pop('d', 'Not Found')

# Prints the returned value
print(value)

Usar el método pop() con diccionarios anidados

Al igual que en las listas, puedes usar pop() para eliminar elementos en diccionarios anidados. Los diccionarios anidados son diccionarios dentro de otros diccionarios. El siguiente código imprimirá {'x': 7, 'y': 8, 'z': 9}.

# Define a nested dictionary
nested_dict = {
    'a': {'x': 1, 'y': 2, 'z': 3},
    'b': {'x': 4, 'y': 5, 'z': 6},
    'c': {'x': 7, 'y': 8, 'z': 9}
}

# Pop a key-value pair from the outer dictionary
removed_dict = nested_dict.pop('c')
print(removed_dict)

También puedes eliminar un valor interno de un diccionario concreto dentro de un anidado indicando su clave. El código de abajo imprime 2.

# Define a nested dictionary
nested_dict = {
    'a': {'x': 1, 'y': 2, 'z': 3},
    'b': {'x': 4, 'y': 5, 'z': 6},
    'c': {'x': 7, 'y': 8, 'z': 9}
}

# Pop a key-value pair from an inner dictionary
popped_inner_value = nested_dict['a'].pop('y')
print(popped_inner_value)

Cómo gestionar el KeyError en diccionarios de Python

Otro enfoque habitual para gestionar el KeyError de forma más elegante es usar un bloque try-except para capturar el error. El siguiente ejemplo muestra cómo usar la técnica try-except con el método pop().

# Create a dictionary
my_dict = {'a': 1, 'b': 2, 'c': 3}

# Create the try-except block to catch the error
try:
    value = my_dict.pop('d')
except KeyError:
    print("Key not found")

También puedes usar la palabra clave in para comprobar si una clave concreta existe en un diccionario antes de eliminarla. El código de abajo usa este método y devuelve Key not found.

# Create a dictionary
my_dict = {'a': 1, 'b': 2}

# Check key existence before popping
key = 'c'
if key in my_dict:
    value = my_dict.pop(key)
    print(f"Popped value: {value}")
else:
    print("Key not found.")

Aspectos de rendimiento

En esta sección, veremos las implicaciones de rendimiento de usar pop() con listas y diccionarios.

Operaciones con listas

La complejidad temporal de las operaciones pop() en listas depende del índice del elemento que se elimina. Cuando usas pop() sin argumento o con -1 como índice para quitar el último elemento, opera en tiempo O(1) porque solo elimina el último elemento de la lista sin cambiar los demás.

Cuando usamos pop() para eliminar el primero o cualquier otro elemento, funciona en O(n) porque implica eliminar un elemento y desplazar los demás a un nuevo orden de índices. Consulta nuestro tutorial Analyzing Complexity of Code through Python para aprender más sobre complejidad temporal en Python.

Operaciones con diccionarios

El método pop() en diccionarios es muy eficiente gracias a la implementación de la tabla hash subyacente. En este caso, pop() con diccionarios suele ser una operación O(1).

Esta complejidad se debe a que el diccionario almacena las claves en una tabla hash y requiere cierto tiempo para inserciones, borrados y búsquedas. Ten en cuenta también que siempre debes especificar la clave a eliminar de un diccionario cuando uses pop(). Consulta nuestro tutorial A Guide to Python Hashmaps para saber más sobre tablas hash.

Alternativas al método pop() de Python

Aunque pop() es eficaz para eliminar elementos de listas y diccionarios, existen algunas alternativas.

Usar la palabra clave del

La palabra clave del se usa para eliminar un elemento de una lista indicando el índice del elemento. del también puede usarse para eliminar pares clave-valor concretos de diccionarios.  Ten en cuenta que al usar del no se devuelve el elemento eliminado, a diferencia de pop(). 

Los siguientes ejemplos muestran cómo usar del para eliminar elementos de una lista y de un diccionario. Tras eliminar el elemento en el índice 2, el siguiente código imprime la nueva lista [1, 2, 4, 5].

# Initialize a list with integers from 1 to 5
my_list = [1, 2, 3, 4, 5]

# Remove item at index 2
del my_list[2]
print(my_list)

El siguiente código imprimirá {'a': 1, 'c': 3}.

# Create a dictionary
my_dict = {'a': 1, 'b': 2, 'c': 3}

# Remove the key 'b' and its associated value from the dictionary
del my_dict['b']
print(my_dict)

Usar el método remove()

El método remove() se aplica solo a listas cuando quieres eliminar un elemento por su valor exacto en lugar de por índice. Este método elimina la primera aparición del elemento indicado si aparece más de una vez en la lista. El código de abajo imprime [1, 3, 4, 5].

# Initialize a list with integers from 1 to 5
my_list = [1, 2, 3, 4, 5]

# Remove the element with the value 2 from the list
my_list.remove(2)
print(my_list)

Conclusión

El método pop() es una función muy sencilla y versátil en Python para eliminar elementos de listas y diccionarios, con multitud de aplicaciones prácticas. Este método es especialmente útil al trabajar con listas y diccionarios porque devuelve el elemento eliminado, que a menudo es justo el comportamiento que necesitas.

Como analista de datos, te recomiendo explorar los itinerarios de habilidades Python Fundamentals y Python Programming de DataCamp para llevar tus habilidades de Python para análisis de datos al siguiente nivel. También te animo a revisar el itinerario profesional Data Analyst with Python para que sigas desarrollando competencias prácticas y alineadas con el mercado laboral.