def quick_sort(array):
if len(array) <=1:
return array
else:
pivot = array[0]
less = [i for i in array[1:] if i <= pivot]
greater = [i for i in array[1:] if i > pivot]
return quick_sort(less) + [pivot] + quick_sort(greater)
2、调用函数
import numpy as np
def main():
# 初始化数组
l = [3, 7, 1, 6, 4, 5, 8, 2]
a = np.array(l)
# 调用快排函数
result = quick_sort(a)
# 输出结果
print(result)
if __name__ == "__main__":
main()