Попытка перекодировать столбец в зависимости от того, содержит ли он определенную строку

Я пытаюсь обобщить группы, и у меня есть столбец заголовков строк в фрейме данных. Я хочу разбить их в зависимости от того, содержат ли они строку. Групп много, но все они содержат строку "Alpha""Beta" или "Gamma". Я хотел бы перекодировать новый столбец с этими заголовками. Представленная ниже функция, похоже, не работает так, как я предполагал, и я не уверен, что лучше - регулярное выражение или подход np.where. Любая помощь приветствуется.

def recode_func(n):
    for i in n:
        if "Alpha" in i:
            i == "Alpha"
        elif "Beta" in i:
            i== "Beta"
        elif "Gamma" in i:
            i == "Gamma"
        else:
            continue