python 快速排序

import random


def quicksort(array):
    if len(array) < 2:
        return array
    else:
        pivot_index = 0
        pivot = array[pivot_index]
        less_part = [
            i for i in array[pivot_index + 1:] if i <= pivot
        ]
        great_part = [
            i for i in array[pivot_index + 1:] if i > pivot
        ]
        return quicksort(less_part) + [pivot] + quicksort(great_part)


if __name__ == ‘__main__‘:
    array = [random.randint(1, 20) for i in range(10)]
    s_a = quicksort(array)
    print(s_a)