Come scrivere una funzione lambda che è condizionata da due variabili (colonne) in python

Question

Ho un set di dati, df, con due variabili, xey. Voglio scrivere una funzione che esegue le operazioni seguenti:

x se x> 100 e y < 50 altro y

Sono abituato a fare l'analisi dei dati in STATA quindi sono relativamente nuovo per panda per l'analisi dei dati. Se aiuta, in Stata sarebbe simile:

sostituire x = cond (x> 100 & y < 50, x, y)

In altre parole, la funzione è subordinata a due colonne in df e restituiscono un valore da una variabile o dall'altra in ogni riga a seconda che la condizione sia soddisfatta.

Finora mi hanno creato nuove variabili attraverso nuove funzioni come:

df.dummyVar = df.x.apply (lambda x: 1 se x> 100 altro 0)

Utilizzando StackOverflow e la documentazione, sono riuscito solo a scoprire come applicare una funzione dipendente da una singola variabile a più di una colonna (usando l'opzione asse). Per favore aiuto.

Answer 1

Uso where:

df['dummyVar '] = df['x'].where((df['x'] > 100) & (df['y'] < 50), df['y']) 

Questo sarà molto più veloce di eseguire un'operazione di applicazione in quanto è Vectorised.

Answer 2

Ti piace questa:

f = lambda x, y: x if x>100 and y<50 else y 

Lambda (s) in Python sono equivalenti ad una definizione di funzione normale.

def f(x, y): 
    return x if x>100 and y<50 else y 

NB: Il corpo di una Lambda deve essere un'espressione valida. Questo significa che non puoi usare cose come: return per esempio; a Lambda restituirà l'ultima espressione valutata.

Per qualche buona lettura vedi:

• Defining Functions
• Lambdas