在字符串中查找特定字符

Question

我正在尝试创建一个接受字符串输入的函数，并检查其中是否包含“=”或“>”，然后在“=”上将内置字符串方法“strip”应用于其后返回一个列表“>”。 为了做到这一点，我正在写一个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