Pandas3

Calculate Special Bonus.py

@@ -0,0 +1,9 @@
+import pandas as pd
+
+def calculate_special_bonus(employees: pd.DataFrame) -> pd.DataFrame:
+    employees['bonus']=employees.apply(
+        lambda x:x['salary']
+        if x['employee_id'] %2 and 
+        not x['name'].startswith('M')
+        else 0 , axis =1)
+    return employees[['employee_id','bonus']].sort_values(by=['employee_id'])

Fix names in a table.py

@@ -0,0 +1,7 @@
+import pandas as pd
+
+def fix_names(users: pd.DataFrame) -> pd.DataFrame:
+
+    users['name'] =users['name'].str[0].str.upper() +users['name'].str[1:].str.lower()  
+
+    return users.sort_values(by=['user_id'])

Patients with a condition.py

@@ -0,0 +1,6 @@
+import pandas as pd
+
+def find_patients(patients: pd.DataFrame) -> pd.DataFrame:
+    df= patients[patients['conditions'].str.startswith('DIAB1') |
+                    patients['conditions'].str.contains(' DIAB1')]
+    return df