代码简化技巧--if/else

415 阅读1分钟

1.省略else

原始:
def do1(input):
    if input > 5:
        print("计划!")
    else:
        print("旅行!")
简化:
def do(input):
    if input > 5:
        print("计划!")
        return
    print("旅行")

2.价值分配

def do(input):
    if input == 1:
        return "男人"
    elif input == 2:
        return "女人"
    return "人妖"

3.前提条件检查

def do(input):
    '''输入必须是0或者1'''
    if input < 0 or input > 1:  # 检查先决条件
        return "人妖"
    return "男人" if input == 0 else "女人"