python中函数对象多重选择代码优化

• 一，函数内存中的存储

• 二，函数对象

• 三，函数中的多个选择代码优化

---

一，函数内存中的存储

我们创建函数能够很好的优化和简洁我们的代码可读性，那么对于函数在内存中的存储又是否清楚呢？如同C语言中一样，函数名在内存中的存储的其实是一个内存地址，当在函数名的后面加上()后就表示需要调用这个函数了。

def func():
    print('from func')

print(func)

例如，这样的一个代码，执行的结果是这样的， 

 所以，通过例子可以看到函数名在内存中存储的时一个内存地址。

二，函数对象

上面我们知道了函数在内存中的存储是以内存地址来存储的，所以函数对象的精髓在于可以将函数名当作一个变量来使用。例如：

def func():
    print('from func')
    

def foo(x):
    print(x)
    return x

func()
res = foo(func)

执行结果就是这样的，

 直接将函数func当作一个变量直接传给函数foo当做函数foo的参数。

那么是否可以这样来玩呢？既然函数名当作函数参数传给的另一个函数，同时函数名在内存中是以内存地址来存储的，在函数名后面加上()后就可以调用函数体，那么是否是可以通过这个返回值res后面加上() 直接来调用函数func的功能呢？

既然有想法，那么就直接来试验一下。

def func():
    print('from func')
    

def foo(x):
    print(x)
    return x

func()
res = foo(func)
res()

尝试的结果表明，既然函数名可以当作变量来使用，那么就可以通过返回值res来直接调用函数 。当明确这些之后就可以玩点花样了。

三，函数体中多个选择代码优化

if... else或者是if... elif ... elif ... else为我们的代码提供了可以选择的方案，提高了代码的灵活性，但是同时也存在一个问题，当存在三四个选择倒还好，至少还能敲出来，但是当有100多个选择的时候该怎么办呢？难道真的要一个一个的选择敲出来吗?那未免代码也太冗长了。而且可读性也不是很高，遇见这样的问题的话就可以通过函数名在内存中的存储来解决。

下面通过一个例子来进行解释，同时加强理解。

def login():
    print('登录功能')

def transfer():
    print('转账功能')

def check_banlance():
    print('查询余额功能')

def withdraw():
    print('提现功能')

dic_func={
    '0':['退出',None],
    '1':['登录功能',login],  # 注意这里不能填写为login()，login()表示开始调用函数
    '2':['转账功能',transfer],
    '3':['查询余额功能',check_banlance],
    '4':['提现功能',withdraw]
}

while True:
    for k in dic_func:
        print(k,dic_func[k][0])

    choice = input('请输入选择功能选项：').strip()  # 防止用户输入选项中存在空格键而导致出错

    if not choice.isdigit(): # 当用户输入的并不是数字的时候
        print("输入格式不对")
        continue 

    if choice == '0':
        break
    
    if choice in dic_func:
        dic_func[choice][1]()  # 执行对应的函数

    else :
        print("请输入正确的功能选项")

执行结果在这并不演示了，想看只想结果的话可以将代码复制，然后在自己的编译器中实验看看。 

这样就简化了选项，当需要添加选项的时候也可以极小的改动，只需要设计好函数，然后在字典中添加好相应的键值对就可以直接添加选项了，这样既优化了代码，同时使代码可读性提高了。