Creé el siguiente marco de datos (que en realidad tiene más de 1000 filas y más de 20 columnas):
d = {'col1': [0, 0, 4, 6], 'col2': [3, 4, 0, 0], 'col3': [0, 10, 0, 0], 'END': [0, 0, 0, 0]} df = pd.DataFrame(data=d) print(df) Out: col1 col2 col3 END 0 0 3 0 0 1 0 4 10 0 2 4 0 0 0 3 6 0 0 0Ahora quiero crear un bucle para cada columna por índice para hacer lo siguiente: verificar si el elemento actual en la columna es mayor que 0 y, en caso afirmativo, si el elemento en la columna a la derecha es igual a 0. Si ambos son verdaderos, luego reste 1 del elemento en la columna y pase a la siguiente columna y repita.
Hasta ahora, mi mejor intento fue con un ciclo while para encontrar los campos correctos:
Counter = len(list(df)) i = 0 while 0 < Counter: if df.iloc[:,i] > 0 and df.iloc[:,i+1] == 0: df.iloc[:,i] - 1 i = i +1Sin embargo, este código genera un error de valor.
Mi resultado deseado se vería así:
col1 col2 col3 END 0 0 2 0 0 1 0 4 9 0 2 3 0 0 0 3 5 0 0 0Esto debería ser bastante rápido:
df[df.ne(0) & df.shift(-1, axis=1).eq(0)] -= 1Producción:
>>> df col1 col2 col3 END 0 0 2 0 0 1 0 4 9 0 2 3 0 0 0 3 5 0 0 0Otra opción
df.sub(df.shift(-1, axis=1).eq(0)).clip(0) col1 col2 col3 END 0 0 2 0 0 1 0 4 9 0 2 3 0 0 0 3 5 0 0 0