挿入ソート(Python)

insertation_sort.py

# -*- coding:utf-8 -*-


def insertation_sort(arr):
    for i in range(1, len(arr)):
        j = i
        while(j > 0 and arr[j - 1] > arr[j]):
            tmp = arr[j - 1]
            arr[j - 1] = arr[j]
            arr[j] = tmp
            j = j - 1
    return arr


if __name__ == '__main__':
    arr = [3, 9, 6, 1, 2]
    print (arr)
    arr = insertation_sort(arr)
    print (arr)