在Python中编写一系列if语句的有效方法？

Question

我有一段代码想重写以使其更有效。我正在使用Python工作，并且没有像Java中那样的switch-break语句，但我觉得类似的结构可以工作。我特别谈论的代码是：def update(arm, array1, array2, array3, array4, array5):    # If the array is empty, add a zero    if not array1:        array1.append(0)    if not array2:        array2.append(0)    if not array3:        array3.append(0)    if not array4:        array4.append(0)    if not array5:        array5.append(0)一般来说，有什么好方法可以避免在 Python 中编写一系列这样的 if 语句，同时还遍历并检查每个数组的特定函数？

Answer 1

您可以更改该函数以接受任意数量的数组*args，然后使用for循环：

def update(arm, *arrays):

    for index, array in enumerate(arrays):

        if not array:

            array.append(0)

        if arm != index:

            array.append(array[-1])

不仅它更短，它还适用于传递给 的任意数量的数组update，因此 、 等中的任何一个都update(arm, array1)可以update(arm, array1, array2)工作。

如果由于某种原因您无法更改函数的签名，您仍然可以使用循环，但必须对循环迭代的数组进行硬编码。

def update(arm, array1, array2, array3, array4, array5,

           array6, array7, array8, array9, array10):

    for index, array in enumerate(array1, array2, array3, array4, array5,

                  array6, array7, array8, array9, array10):

        if not array:

            array.append(0)

        if arm != index:

            array.append(array[-1])

Answer 2

您可以为任何想要为空列表分配零元素的函数定义一个装饰器。这将允许您添加初始化，而无需更改函数内的任何内容。

def noEmptyLists(func):

    def decorated(*args,**kwargs):

        for a in (*args,*kwargs.values()):

            if isinstance(a,list) and not a:

                a.append(0)

        return func(*args,**kwargs)

    return decorated

@noEmptyLists

def update(arm, array1, array2, array3, array4, array5):

    print(arm, array1, array2, array3, array4, array5)

update(1,[],[],[],[],[])

# 1 [0] [0] [0] [0] [0]