pythonic方式重用样板循环结构而不调用if子句n次

Question

我继承了一个遗留代码库，里面有很多嵌套的 for 循环，看起来像：def func(infile, some_other_data, outfile, status_variable):    with open(infile, 'r') as f:        with open(outfile, 'w') as outf:            for line in f:                # parse line                for element in some_other_data:                    standard_function(line, element)                    if status_variable == 'status_A':                        function_A(line, element)                    elif status_variable == 'status_B':                        function_B(line, element)                    # handle other possible status variables                    outf.write(new_line)此代码与性能相关。为了加快速度（除了其他更改之外），我想摆脱所有被调用 n*m 次的 if 子句，测试表明这确实提供了 10% 的改进。为此，我简单地为每个可能的状态变量复制并修改了主循环函数，并相应地调用了不同的函数。这有效地将 if 子句移到了循环之外。但是，它非常丑陋，并且使库变大了 4 倍。是否有一种（相当）简单的 pythonic 方法来处理这种情况，我想重用样板循环，只是改变每次迭代所做的事情，而不是每次都处理条件？我一直在玩装饰器，根据状态变量动态返回循环函数调用不同的子函数，但从可读性的角度来看，最终结果看起来很糟糕。我绝不是 Python 专家，所以我可能会忽略一些在这里可能有用的方便的高级功能。任何建议都非常感谢。

Answer 1

理想情况下，您将传递函数本身而不是状态变量，但由于这是遗留代码，因此不更改接口的一种解决方案是设置函数字典，如下所示：

def func(infile, some_other_data, outfile, status_variable,

         status_functions={

             'status_A': function_A,

             'status_B': function_B,

         }

        ):

    try:

        status_function = status_functions[status_variable]

    except KeyError:

        status_function = lambda line, element: None

    with open(infile, 'r') as f, open(outfile, 'w') as outf:

        for line in f:

            # parse line

            for element in some_other_data:

                standard_function(line, element)

                status_function(line, element)

                # handle other possible status variables

                outf.write(new_line)

Answer 2

如果status_variable-> function_name之间有直接对应关系，并且所有调用都是常规调用：function(line, element)您可以传入函数：

def func(infile, some_other_data, outfile, function_from_status_variable):

    with open(infile, 'r') as f:

        with open(outfile, 'w') as outf:

            for line in f:

                # parse line

                for element in some_other_data:

                    standard_function(line, element)

                    function_from_status_variable(line, element)

                    outf.write(new_line)

计算一次，因此：

def calc_function(status_variable):

    if status_variable == 'status_A':

        return function_A

    elif status_variable == 'status_B':

        return function_B

    # other tests follow, plus handle an unknown value

最后像这样调用函数：

function = calc_function(status_variable)

func(infile, some_other_data, outfile, function)