如何在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实现以及使用例子。根据不同的数据规模和性能要求,选择合适的排序算法能够提高程序的执行效率。
