es.davy.ai

Preguntas y respuestas de programación confiables

¿Tienes una pregunta?

Si tienes alguna pregunta, puedes hacerla a continuación o ingresar lo que estás buscando.

Python pandas cómo usar una función con math, usar apply y lambda

13 Jul, 2023 Programación

Tengo una parte de programa con el siguiente dataframe de ejemplo:

import pandas as pd
from math import floor

d = {'ind': ['a', 'b', 'c'], 'col1': [1, 2, 3], 'col2': [4, 5, 6], 'col3': [7, 8, 9], 'spec': [9, 6, 3]}
df = pd.DataFrame(data=d).set_index('ind')

def func(x, y):
    return (1-x)*(y+1)*0.9

print(df)

df2 = df.apply(lambda x: func(x, df.spec))

print(df2)

Funciona bien, pero cuando modifico un poco la función y le agrego floor:

def func(x, y):
    return floor((1-x)*(y+1)*0.9)

Obtengo el siguiente error:

TypeError: cannot convert the series to <class 'float'="">

¿Cómo puedo cambiar df2 o func para que funcione?

EDICIÓN DESPUÉS DE UNA BUENA RESPUESTA:

Ahora veo que olvidé que tengo una condición en la función, y necesito algo más para que funcione.

def func(x, y):
    if x == 1:
        return y
    else:
        return floor((1-x)*(y+1)*0.9)

Con esta función y las respuestas que recibí, obtengo:

ValueError: The truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all().

Señala la línea if x == 1:.

Tags:  , , , ,

Answer

  1. Avatar for davy.ai
    davy.ai

    El problema es que la función floor no se puede aplicar directamente a una serie de datos. Una solución para esto es usar el método applymap del objeto DataFrame para aplicar la función elemento a elemento a cada celda:

    import pandas as pd
from math import floor

d = {'ind': ['a', 'b', 'c'], 'col1': [1, 2, 3], 'col2': [4, 5, 6], 'col3': [7, 8, 9], 'spec': [9, 6, 3]}
df = pd.DataFrame(data=d).set_index('ind')

def func(x,y):
    return (1-x)*(y+1)*0.9

df2 = df.apply(lambda x: func(x, df.spec))
df3 = df2.applymap(lambda x: floor(x)) # aplica floor elemento a elemento

print(df)
print(df3)

    Esto aplicará la función floor a cada celda del DataFrame df2, que contiene la salida de la función func. El método applymap aplica una función elemento a elemento a cada celda del DataFrame, lo cual resuelve el problema de aplicar la función floor a una serie de datos.

    Sin embargo, ten en cuenta que esta solución solo funcionará si la función func es capaz de devolver un valor flotante para cada celda. Si la función devuelve un valor booleano o una cadena de texto para algunas celdas, podrías encontrarte con problemas.

Comments are closed.