
SAS2PYTHON - Logik & IF-Anweisungen
Posted on 19 July 2022 at 17:00 by Stefan Weingaertner
In der Episode 15 unseres heutigen Blog-Beitrags zeigen wir, wie man in SAS/BASE und in Pandas IF-Anweisungen anwenden kann und das Ergebnis jeweils in Form einer neuen Spalte abspeichert.
Ein Pandas DataFrame ist einem SAS Data Set sehr ähnlich - es handelt sich um eine zweidimensionale tabellarische Datenstruktur mit gelabelten Spalten, die von unterschiedlichem Datentyp sein können. Wie in diesem Beitrag gezeigt wird, kann nahezu jede Operation, die mit dem SAS Data Step auf einen Datensatz angewendet werden kann, auch in Pandas ausgeführt werden.
Input: Unsere Input-Tabelle besitzt 6 Datensätze und besteht aus den folgenden Spalten.
Name Department Salary
Vincent Vega Sales 5400
Marsellus Wallace Marketing 3200
Butch Coolidge HR 3100
Honey Bunny IT 6000
Esmeralda Villalobos Purchasing 1700
Jules Winnfield Finance 2500
Output: Wir fügen der Tabelle durch die Anwendung der IF/ELSE-Anweisung eine neue Spalte hinzu:
- Spalte "Range" durch die logische Abfrage der Spalte "Salary".
Da manche Operationen in SAS und Python unterschiedlich ausgeführt werden, generieren die beiden Programmiersprachen auch leicht unterschiedliche Ausgabe-Formate. Wir haben daher die Outputs mit dem dazugehörigen Code für SAS und Python separat aufgeführt.
SAS-Code: Das Berechnen der Gehalts-Stufe auf Basis der Spalte "Salary" würde man für ein SAS Dataset "df" wie folgt umsetzen.
data df;
set df;
IF Salary < 2000 THEN Range = "low";
ELSE IF 2000 <= Salary <= 4000 THEN Range = "medium";
ELSE IF 4000 < Salary THEN Range = "high";
run;
SAS-Output: Das Ergebnis sieht in SAS wie folgt aus, dem SAS Data Set wird die neue Spalte "Salary" hinzugefügt.
Obs Name Department Salary Range
1 Vincent Vega Sales 5400 high
2 Marsellus Wallace Marketing 3200 medium
3 Butch Coolidge HR 3100 medium
4 Honey Bunny IT 6000 high
5 Esmeralda Villalobos Purchasing 1700 low
6 Jules Winnfield Finance 2500 medium
Python-Code: In Python würde man das Ableiten der neuen Spalte "Salary" für ein Pandas Dataframe "df" mit einer Funktion "assign_range" umsetzen und diese dann wie folgt aufrufen.
# Funktion assign_range
def assign_range(salary):
if salary < 2000:
result = 'low'
elif salary <= 4000:
result = 'medium'
else:
result = 'high'
return result
# Anwenden der Funktion assign_range auf die Spalte "Salary"
import pandas as pd
df['Range'] = df['Salary'].apply(assign_range)
Python-Output: Das Ergebnis sieht in Pandas wie folgt aus, dem Pandas Dataframe wird die neue Spalte "Range" hinzugefügt.
Name Department Salary Range
0 Vincent Vega Sales 5400 high
1 Marsellus Wallace Marketing 3200 medium
2 Butch Coolidge HR 3100 medium
3 Honey Bunny IT 6000 high
4 Esmeralda Villalobos Purchasing 1700 low
5 Jules Winnfield Finance 2500 medium
Wenn Sie die nächste Episode nicht abwarten möchten, können Sie auch gerne unser umfangreiches Training Python für SAS Professionals buchen.
Informationen zu den nächsten Kursterminen und zu unserem gesamten Trainingsangebot finden Sie hier.