用字典代替'if-elif-else'

在实际应用中，我们经常会需要采用if-elif-else控制语句以根据不同条件，作出不同的操作。if-elif-else固然可以，但是它也存在冗余的缺点，特别是当条件较多时这一缺点尤为明显。因此，本文考虑采用字典来处理这一情况。

Python的字典是由键值对构成，键是唯一的，并且每个键都有唯一的值对应。因此，我们可以将条件控制语句中的条件作为键，将相应条件下的操作作为值。如果“操作”较为复杂时，我们可以将其打包为一个函数，并将此函数作为字典的“值”。

举例如下：

import numpy as np

np.random.seed(1234)

true_label = np.random.choice(np.array([0,1]),

                              size=10000, p = [0.5, 0.5],

                              replace=True)

pred_prob = np.random.rand(10000)

metric = 'precision'

perform = np.zeros_like(pred_prob)

for j, thres in enumerate(pred_prob):

    pred_label = np.where(pred_prob>=thres, 1, 0)

    if metric == 'recall':

        perform[j] = np.logical_and(pred_label, true_label).sum()/true_label.sum()

    elif metric == 'precision':

        perform[j] = np.logical_and(pred_label, true_label).sum()/pred_label.sum()

    elif metric == 'accuracy':

        perform[j] = 1 - np.logical_xor(pred_label, true_label).sum()/pred_label.shape[0]

上述代码是利用条件控制语句，根据需要计算"召回率"、"精准率"、“准确率”，具体需要即由变量metric控制。与之对应的字典版本即为

def recall(pred_label, true_label):

    return np.logical_and(pred_label, true_label).sum()/true_label.sum()

def precision(pred_label, true_label):

    return np.logical_and(pred_label, true_label).sum()/pred_label.sum()

def accuracy(pred_label, true_label):

    return 1 - np.logical_xor(pred_label, true_label).sum()/pred_label.shape[0]

metric_dict = {'recall'   : recall,

               'precision': precision,

               'accuracy' : accuracy}

metric = 'precision'

evalua = np.zeros_like(pred_prob)

for i, thres in enumerate(pred_prob):

    pred_label = np.where(pred_prob>=thres, 1, 0)

    evalua[i]  = metric_dict.get(metric, None)(pred_label, true_label)

可以看到，条件控制语句中的操作被打包成了函数，并作为字典中的值。接下来根据键，对字典的值进行索取即可，而这里的键值即为条件，也由变量metric控制。

巴特西

用字典代替'if-elif-else'

最新文章

热门文章