My code

1
2
3
4
5
import pandas as pd

def find_patients(patients: pd.DataFrame) -> pd.DataFrame:
    patients = patients.loc[patients['conditions'].str.contains(' DIAB1') | patients['conditions'].str.startswith('DIAB1'),['patient_id','patient_name','conditions']]
    return patients

other code

1
2
3
4
5
6
7
8
9
10
import pandas as pd

def find_patients(patients: pd.DataFrame) -> pd.DataFrame:
    # Use the str.contains() method to find patients with Type I Diabetes
    patients_with_diabetes = patients[patients['conditions'].str.contains(r'\bDIAB1')]
    
    # Select only the required columns
    result_df = patients_with_diabetes[['patient_id', 'patient_name', 'conditions']]
    
    return result_df
1
2
def find_patients(patients: pd.DataFrame) -> pd.DataFrame:
    return patients[patients['conditions'].str.contains(r'(^DIAB1)|( DIAB1)')]