有没有办法重写 if 语句以避免重复这两行代码？

Question

我写了 insert_sort function 并添加了一个 if 语句来选择是否要以升序或降序模式对数组进行排序。 正如您在 if 语句中的以下代码中看到的那样，重复了 2 行代码。 我想知道是否有另一种方法来编写这些如果，所以没有重复的行。 我一直在考虑它，但我想不出任何替代方案。

提前致谢。

def swap_elements(array, pos1, pos2):
    array[pos1], array[pos2] = array[pos2], array[pos1]
    return array

def insertion_sort(array, method):
    i = 1
    while i < len(array):
        key = array[i]
        j = 0
        while j < i:
            if method:
                if array[j] > key:
                    key = array[j]
                    array = swap_elements(array, i, j)
            else:
                if array[j] < key:
                    key = array[j]
                    array = swap_elements(array, i, j)
            j += 1
        i += 1

    return array

Answer 1

最新版本的代码：

def swap_elements(array, pos1, pos2):
    array[pos1], array[pos2] = array[pos2], array[pos1]

def insertion_sort(array, method):
    i = 1
    while i < len(array):
        key = array[i]
        j = 0
        while j < i:
            if (method and array[j] > key) or (not method and array[j] < key):
                key = array[j]
                swap_elements(array, i, j)
            j += 1
        i += 1