能否将函数返回值赋值给变量？附邮箱域名判断代码需求

关于函数返回值赋值给变量 #

当然可以把函数的返回值赋值给变量，用法和普通变量赋值一样，举个例子：

# 调用函数并把返回值存到变量里
result = mail("test_user@hotmail.com")
print(result)  # 输出 "Yes"

你的代码修正方案 #

你写的代码里的核心错误是：在给变量x赋值时错误使用了return关键字——return是用来从函数返回结果的，不能放在赋值语句里。下面是修正后的正确代码：

def mail(var):
    # 直接拆分邮箱字符串，提取@后面的域名部分
    x = var.split('@')[1]
    if x == 'hotmail.com':
        return 'Yes'
    else:
        return 'No'

可选的健壮性优化（处理非法邮箱） #

如果需要应对没有@符号的非法邮箱格式，可以加个判断避免程序报错：

def mail(var):
    if '@' not in var:
        return 'Invalid email'
    x = var.split('@')[1]
    # 用三元表达式简化判断逻辑
    return 'Yes' if x == 'hotmail.com' else 'No'

内容的提问来源于stack exchange，提问作者Grzegosh