问题#

def func(parameter):
    if parameter == '0001':
        return '小明'
    elif parameter == '0002':
        return '小红'
    elif parameter == '0003':
        return '小马'
    elif parameter == '0004':
        return '晴晴'
    elif parameter == '0005':
        return '纳米'
    else:
        raise NameError('未查询到此人')

重构以上代码，必须保证其功能一致。

分析问题#

此问题使用了 if...elif...else 来尝试查询键值对数据，不仅低效繁琐，还可读性差，使用字典结构就能优雅高效地存放和查询此类数据。

另外，数据与逻辑要区分开，这样逻辑就能更方便地统一处理，数据也能统一修改。

对于根据键获取值的方式，可以使用索引，也可以使用 get() 方法，但使用 get() 方法更加简洁些。

重构#

先将数据转到字典：

def func(parameter):
    base = {
        '0001': '小明',
        '0002': '小红',
        '0003': '小马',
        '0004': '晴晴',
        '0005': '纳米'
    }

然后编写查询逻辑，只需要 3 行：

def func(parameter):
    base = {
        '0001': '小明',
        '0002': '小红',
        '0003': '小马',
        '0004': '晴晴',
        '0005': '纳米'
    }
    if name := base.get(parameter):
        return name
    raise NameError('未查询到此人')

需要注意的是，这里使用了:= 运算符，既能赋值又可以条件判断，非常地好用。（赞赏）