字符串变为函数

    今天看到一段代码,在一字典中有个字段为gt,gte,lt,lte,然后根据此字段进行判断,例子代码如下:

dict_test1 = {"A":2,"B": 3, "rule":"gt"}
dict_test2 = {"A":2,"B": 3, "rule":"gte"}
dict_test3 = {"A":2,"B": 3, "rule":"lt"}
# 需要对最后的rule进行判断,然后判断前面A,B谁更大,然后进行其他流程操作
if dict_test1['rule'] == "gt":
    if A > B :
        control(1)
    else:
        control(2)
elif dict_test1['rule'] == "gte":
    if A >= B :
        control(1)
    else:
        control(2)
elif dict_test1['rule'] == "lte":
    if A <= B :
        control(1)
    else:
        control(1)
elif dict_test1['rule'] == "lt":
    if A < B :
        control(1)
    else:
        control(2)

看到这样的代码,感觉太麻烦了,特别是如果第二层判断后,如果代码重复,那完全就是消耗性能,以及冗余的代码。

优化后:

from operator import lt, gt, ge, le

dict_test1 = {"A":2,"B": 3, "rule":"gt"}
dict_test2 = {"A":2,"B": 3, "rule":"ge"}
dict_test3 = {"A":2,"B": 3, "rule":"lt"}
dict_test4 = {"A":2,"B": 3, "rule":"le"}
if locals()[dict_test1["rule"]](A,B):
    print("A", dict_test1["rule"], "B")
    pass
else:
    pass

主要优化:1 将rule中的gte,lte需要改为le(小于等于),ge(大于等于)
                   2 多个步骤优化成一个,自动进行判断。无需其他
注意点:

  1. 需要提前将相关函数import进来
  2. 转化方式:locals()[func](**arg)
  3. 通过globals 也是可以的,看个人需求,方式和locals一样


 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值