Wie schreibe ich eine Lambda-Funktion, die in Python von zwei Variablen (Spalten) abhängig ist

Question

Ich habe einen Datensatz, df, mit zwei Variablen, x und y. Ich möchte eine Funktion schreiben, macht folgendes:

x wenn x> 100 und y < 50 sonst y

Ich bin es gewohnt, Datenanalyse in STATA Dabei bin ich relativ neu Pandas für die Datenanalyse. Wenn es hilft, in Stata würde es wie folgt aussehen:

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

Mit anderen Worten, die Funktion ist davon abhängig zwei Spalten in df und gibt einen Wert von einer Variablen oder der anderen in jeder Zeile zurück, abhängig davon, ob die Bedingung erfüllt ist.

Bisher habe ich wie neue Variablen durch neue Funktionen wurden zu erstellen:

df.dummyVar = df.x.apply (lambda x: 1, wenn x> 100 sonst 0)

Mit StackOverflow und der Dokumentation konnte ich nur herausfinden, wie man eine Funktion in Abhängigkeit von einer einzelnen Variablen auf mehr als eine Spalte anwendet (mit der Option axis). Bitte helfen Sie.

Answer 1

Verwendung where:

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

Dies wird schneller sein viel als eine Operation anwenden, wie es ist vektorisiert.

Answer 2

So:

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

Lambda (s) in Python ist äquivalent zu einer normalen Funktionsdefinition.

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

NB: Der Körper eines Lambda muss ein gültiger Ausdruck sein. Dies bedeutet, dass Sie keine Dinge wie: return zum Beispiel verwenden können; Ein Lambda gibt den letzten ausgewerteten Ausdruck zurück.

Für einige gute Lektüre siehe:

• Defining Functions
• Lambdas