为何return函数未将变量传递给下一个函数？NameError报错排查

问题代码 #

def userInput():
    a = float(input("Enter a: "))
    b = float(input("Enter b: "))
    return (a,b)

def printFunction(a2,b2):
    print(a2)
    print(b2)

def main():
    userInput()
    printFunction(a2,b2)
    
main()

报错信息 #

NameError: 未定义名称 'a2'

错误原因 #

你在userInput()中正确返回了包含a和b的元组，但在main()里调用userInput()时，没有将返回结果赋值给任何变量。a2和b2这两个名称在main()的作用域中完全不存在，因此触发了名称错误。

修正方案 #

不需要将input移出userInput函数，只需在main()中正确接收userInput()的返回值即可，有两种常用方式：

方式1：解包返回值并赋值给变量 #

def userInput():
    a = float(input("Enter a: "))
    b = float(input("Enter b: "))
    return (a,b)

def printFunction(a2,b2):
    print(a2)
    print(b2)

def main():
    a2, b2 = userInput()  # 接收返回的元组并解包为两个变量
    printFunction(a2,b2)
    
main()

方式2：直接解包元组作为函数参数 #

def userInput():
    a = float(input("Enter a: "))
    b = float(input("Enter b: "))
    return (a,b)

def printFunction(a2,b2):
    print(a2)
    print(b2)

def main():
    printFunction(*userInput())  # 使用*运算符解包元组，直接传递给函数
    
main()

以上两种方式都能在保留userInput函数结构的前提下解决报错问题。

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