插入排序 python

下面是 Python 实现的插入排序代码：

def insertion_sort(arr):
    for i in range(1, len(arr)):
        key = arr[i]
        j = i - 1
        while j >= 0 and key < arr[j]:
            arr[j + 1] = arr[j]
            j -= 1
        arr[j + 1] = key
    return arr

其中，`arr` 是待排序的数组。该算法的时间复杂度为 $O(n^2)$。