Reemplaza loc e iloc para mejorar el rendimiento en bucles

Publicado el 13 febrero 2023 por Daniel Rodríguez @analyticslane

Dos de los métodos más utilizados para acceder a los elementos de una Serie o un DataFrame en Pandas son loc e iloc. Algo que se debe a su gran flexibilidad, permite seleccionar elementos o subconjuntos usando tanto los nombres de las etiquetas ( loc) como la posición ( iloc) de forma intuitiva. Aunque, en algunos casos, pueden ofrecer problemas de rendimiento. Por ejemplo, en los bucles, cuando es necesario acceder a múltiples elementos. Problemas de rendimiento que se pueden solucionar reemplazando loc e iloc por at e iat para operar sobre los elementos de la serie o el DataFrame dentro de un bucle.

Operar sobre las filas de un DataFrame con loc e iloc

Para comprobar los problemas de rendimiento de loc e iloc se puede crear un DataFrame aleatorio con medio millón de filas y tres columnas.

import pandas as pd
import numpy as np

np.random.seed(0)

df = pd.DataFrame(np.random.randint(0, 1000, (500000, 3)), columns=['A', 'B', 'C'])
df.head()
     A    B    C
0  684  559  629
1  192  835  763
2  707  359    9
3  723  277  754
4  804  599   70

Ahora se puede plantear un problema sencillo, sumar los valores de la primera (A) y segunda columna (B), para almacenar el resultado en la tercera (C). Lo que se puede implementar con el siguiente código en el que también se mide el tiempo de ejecución con time().

import time
start = time.time()

for index, row in df.iterrows():
    df.loc[index, 'C'] = row.A + row.B

end = time.time()
print(end - start)
16.9221031665802

En mi equipo el código ha tardado 16,92 segundos en ejecutarse. Se puede repetir el ejercicio cambiando loc por iloc, con lo que el código quedaría de la siguiente forma.

import time
start = time.time()

for index, row in df.iterrows():
    df.iloc[index, 2] = row.A + row.B

end = time.time()
print(end - start)
15.030578851699829

Lo que ofrece una ligera mejora del rendimiento, ahora el proceso ha tardado 15,03 segundos.

Operar sobre las filas de un DataFrame con at e iat

Este problema se puede resolver con at en lugar de loc, solamente reemplazo un método por el otro. Por lo que el código anterior quedaría como

import time
start = time.time()

for index, row in df.iterrows():
    df.at[index, 'C'] = row.A + row.B

end = time.time()
print(end - start)
9.518975973129272

Ahora solamente ha tardado 9,52 segundos en terminar la tarea, lo que supone una mejora del 78% respecto al uso de iloc. En el caso de iat el código que se debería ejecutar sería el siguiente.

import time
start = time.time()

for index, row in df.iterrows():
    df.at[index, 'C'] = row.A + row.B

end = time.time()
print(end - start)
11.972346067428589

Aunque en este caso el tiempo necesario para completar la tarea es de 11,97 segundos, un aumento de rendimiento menor respecto a iloc pero aún más importante.

¿Por qué no siempre se puede reemplazar loc e iloc por at e iat para mejorar el rendimiento en bucles?

Los métodos loc y at no funcionan exactamente igual, por lo que no es posible reemplazar uno por otro en todas las ocasiones. Lo mismo que en el caso de iloc e iat. Mientras que con loc e iloc se puede seleccionar un subconjunto del DataFrame original con at e iat solamente se pueden seleccionar un elemento. Por lo que, cuando se quiere crear un nuevo conjunto de datos solamente con unos valores se tiene que usar obligatoriamente loc o iloc. Por ejemplo, para seleccionar las tres primeras filas del objeto.

df.loc[:2]
     A    B    C
0  684  559  629
1  192  835  763
2  707  359    9

En este caso si se cambia loc por at el código producirá un error.

Mejorar el rendimiento en bucles reemplazado iterrows()

En una publicación anterior se vio que iterrows() no es el método más eficiente para iterar sobre un DataFrame. Iterar sobre los índices con range() es una forma mucho más eficaz. Algo que se puede realizar fácilmente para el problema de ejemplo.

import time
start = time.time()

for _ in range(250):
    for index in range(len(df.head())):
        df.loc[index, 'C'] = df.loc[index, 'A'] + df.loc[index, 'B']

end = time.time()
print(end - start)
0.059928178787231445

Si se repite el ejercicio anterior iterando sobre los índices con loc se tiene que el proceso, replicado esta vez 250 veces, solamente tarda 0,0599 segundos.

import time
start = time.time()

for _ in range(250):
    for index in range(len(df.head())):
        df.at[index, 'C'] = df.at[index, 'A'] + df.at[index, 'B']

end = time.time()
print(end - start)
0.011784791946411133

Pero si se repite con at el tiempo se reduce a solo 0,0118 segundos. Por lo que la mejor ahora es de un factor 5.

Aunque esta es la opción más eficiente para resolver este problema, en las secciones anteriores se empleó `iterorws()` mostrar de una forma más clara al diferencia de rendimiento entre un método y otro.

Conclusiones

En esta entrada se ha visto que cambiar loc por at e iloc por iat, cuando esto es posible, se puede traducir en una importante mejora del rendimiento cuando se trabaja con bucles. Pudiendo mejorar en un factor de 5. Además se ha vuelto a comprobar que iterrows() tampoco es la forma más eficiente de iterar sobre un DataFrame, iterar sobre los índices es mucho más eficaz.