欢迎访问宙启技术站
智能推送

如何在Python中实现基本的数据排序功能

发布时间:2024-01-12 10:55:04

Python提供了多种排序算法,包括冒泡排序、插入排序、选择排序、快速排序、归并排序等。下面将介绍其中几种常用的排序算法,并给出使用例子。

1. 冒泡排序(Bubble Sort)

冒泡排序是一种简单的排序算法,它重复地遍历要排序的列表,比较相邻的两个元素,并通过交换位置来将较大的元素推到末尾。示例代码如下:

def bubble_sort(nums):
    n = len(nums)
    for i in range(n):
        for j in range(0, n-i-1):
            if nums[j] > nums[j+1]:
                nums[j], nums[j+1] = nums[j+1], nums[j]
    return nums

# 使用例子
nums = [5, 2, 9, 1, 3]
sorted_nums = bubble_sort(nums)
print(sorted_nums) # 输出 [1, 2, 3, 5, 9]

2. 插入排序(Insertion Sort)

插入排序的思想是将列表分为已排序和未排序两部分,从未排序部分中选择一个元素并插入到已排序部分的合适位置。示例代码如下:

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

# 使用例子
nums = [5, 2, 9, 1, 3]
sorted_nums = insertion_sort(nums)
print(sorted_nums) # 输出 [1, 2, 3, 5, 9]

3. 选择排序(Selection Sort)

选择排序的思想是在未排序部分找到最小(或最大)的元素并放到已排序部分的末尾。示例代码如下:

def selection_sort(nums):
    n = len(nums)
    for i in range(n):
        min_idx = i
        for j in range(i+1, n):
            if nums[j] < nums[min_idx]:
                min_idx = j
        nums[i], nums[min_idx] = nums[min_idx], nums[i]
    return nums

# 使用例子
nums = [5, 2, 9, 1, 3]
sorted_nums = selection_sort(nums)
print(sorted_nums) # 输出 [1, 2, 3, 5, 9]

4. 快速排序(Quick Sort)

快速排序是一种常用的排序算法,它通过选择一个元素作为基准,将小于基准的元素放在基准的左侧,大于基准的元素放在基准的右侧,然后分别对左右两个部分进行递归排序。示例代码如下:

def quick_sort(nums):
    if len(nums) <= 1:
        return nums
    pivot = nums[len(nums)//2]
    left = [x for x in nums if x < pivot]
    middle = [x for x in nums if x == pivot]
    right = [x for x in nums if x > pivot]
    return quick_sort(left) + middle + quick_sort(right)

# 使用例子
nums = [5, 2, 9, 1, 3]
sorted_nums = quick_sort(nums)
print(sorted_nums) # 输出 [1, 2, 3, 5, 9]

5. 归并排序(Merge Sort)

归并排序通过将列表分成两半递归排序,然后将两个有序子列表合并成一个有序列表。示例代码如下:

def merge_sort(nums):
    if len(nums) <= 1:
        return nums
    mid = len(nums) // 2
    left = merge_sort(nums[:mid])
    right = merge_sort(nums[mid:])
    return merge(left, right)

def merge(left, right):
    result = []
    i = j = 0
    while i < len(left) and j < len(right):
        if left[i] < right[j]:
            result.append(left[i])
            i += 1
        else:
            result.append(right[j])
            j += 1
    result.extend(left[i:])
    result.extend(right[j:])
    return result

# 使用例子
nums = [5, 2, 9, 1, 3]
sorted_nums = merge_sort(nums)
print(sorted_nums) # 输出 [1, 2, 3, 5, 9]

以上是常见的几种排序算法的Python实现以及使用例子。根据不同的数据规模和性能要求,选择合适的排序算法能够提高程序的执行效率。