извлекать строку с помощью нескольких условий в течение петы питона панд

Question

Я хотел бы извлечь строку панд с несколькими условиями, применяя if else в цикле. Тем не менее, это, кажется, не работает и возвращает только первый столбец Любые советы для этого?

|col_a|col_b|
|peter--bob:5067561|peter--bob:5067561|
|chris**bbb:5067561|chris**bbb:5067561|
|bob##ccc:5067561|bob##ccc:5067561|


def get_string(df):

    cols = df.columns[0:20]

    for col in cols:

        if col.find('*') == -1: 
            return df[col].astype(str).str.split('*').str[0]

        if col.find('-') == -1:
            return df[col].astype(str).str.split('-').str[0]

        if col.find('#') == -1:
            return df[col].astype(str).str.split('#').str[0]

Answer 1

В вашем цикле проверяется имя столбца вместо значений столбца. Решение - выбрать столбец по df[col] и проверить по Series.str.contains, применить решение и присвоить обратно DataFrame:

def get_string(df):

    cols = df.columns[0:20]

    for col in cols:
        for v in ['*','-','#']:
            mask = df[col].str.contains(v, na=False, regex=False)
            df.loc[mask, col] = df.loc[mask, col].str.split(v).str[0]

    return df

print (get_string(df))
   col_a  col_b
0  peter  peter
1  chris  chris
2    bob    bob