Поиск определенного символа в строке

Question

Я пытаюсь создать функцию, которая принимает строковый ввод, и проверяет, имеет ли она внутри «=» или «>», а затем возвращает список после применения встроенного строкового метода «полоса» к нему на «=» или «>». Чтобы сделать это, я пишу заявление if и elif.

def split_list_math(a_str): 
    if(contains("=") == True): 
     a_str2 = a_str.split("=") 
    elif(contains(">") == True): 
     a_str2 = a_str.split(">") 
    return a_str2 

Что я могу заменить «содержать» в своем коде, чтобы эта функция работала?

Answer 1

Используйте in оператор:

if "=" in a_str: 

Демо:

>>> '=' in '!@#$%^' 
False 
>>> '=' in '=-0987' 
True