Kết hợp hai danh sách đã sắp xếp trong Python

List là cấu trúc dữ liệu Python được dùng nhiều. Bài viết trình bày cách kết hợp phần tử 2 list và cho ra kết quả đã sắp xếp.

Sử dụng + và sorted

Toán tử + nối 2 list thành một. Áp dụng sorted() để sắp xếp list kết quả.

Ví dụ

listA = ['Mon', 'Tue', 'Fri']
listB = ['Thu', 'Fri', 'Sat']  # Danh sách cho trước
print("Given list A is : ", listA)
print("Given list B is : ", listB)
# Cộng và sắp xếp
res = sorted(listA + listB)  # Kết quả
print("The combined sorted list is : \n", res)

Kết quả

Chạy đoạn mã trên cho kết quả sau:

Given list A is : ['Mon', 'Tue', 'Fri'] Given list B is : ['Thu', 'Fri', 'Sat'] The combined sorted list is : ['Fri', 'Fri', 'Mon', 'Sat', 'Thu', 'Tue']

Sử dụng heapq.merge

Hàm merge() từ module heapq kết hợp 2 list. Sau đó áp dụng sorted() cho kết quả.

Ví dụ

from heapq import merge
listA = ['Mon', 'Tue', 'Fri']
listB = ['Thu', 'Fri', 'Sat']  # Danh sách cho trước
print("Given list A is : ", listA)
print("Given list B is : ", listB)
# Merge
res = list(merge(listA, listB))  # Kết quả
print("The combined sorted list is : \n", sorted(res))

Kết quả

Chạy đoạn mã trên cho kết quả sau:

Given list A is : ['Mon', 'Tue', 'Fri'] Given list B is : ['Thu', 'Fri', 'Sat'] The combined sorted list is : ['Fri', 'Fri', 'Mon', 'Sat', 'Thu', 'Tue']