Thông báo: Download 4 khóa học Python từ cơ bản đến nâng cao tại đây.
Cách chia List thành các phần bằng nhau trong Python
Trong Python, khi làm việc với danh sách (list), đôi khi bạn cần chia chúng thành các phần nhỏ hơn có kích thước bằng nhau để dễ dàng xử lý hoặc thao tác dữ liệu. Có nhiều cách khác nhau để thực hiện điều này một cách hiệu quả và linh hoạt. Bài viết này sẽ giới thiệu những phương pháp phổ biến và hiệu quả nhất để chia một danh sách thành các phần có kích thước bằng nhau, giúp bạn dễ dàng lựa chọn cách phù hợp nhất cho nhu cầu của mình.
Tự viết hàm sinh (generator) trong Python
Bạn có thể tự viết một hàm sinh (generator) để thực hiện việc chia danh sách như sau:
my_list = list(range(10)) def chunk(lst, n): for i in range(0, len(lst), n): yield lst[i:i + n] chunks = list(chunk(my_list, 3)) print(chunks) # Kết quả: [[0, 1, 2], [3, 4, 5], [6, 7, 8], [9]]
Trong đoạn code này, hàm chunk
sẽ lấy các phần từ của danh sách ban đầu my_list
và chia thành các phần có kích thước bằng 3.
Dùng một dòng lệnh (one-liner)
Nếu bạn muốn một cách viết ngắn gọn hơn, có thể sử dụng comprehension của Python:
Bài viết này được đăng tại [free tuts .net]
n = 3 my_list = list(range(10)) # Generator comprehension chunks = (my_list[i:i + n] for i in range(0, len(my_list), n)) print(list(chunks)) # Kết quả: [[0, 1, 2], [3, 4, 5], [6, 7, 8], [9]] # List comprehension chunks = [my_list[i:i + n] for i in range(0, len(my_list), n)] print(chunks) # Kết quả: [[0, 1, 2], [3, 4, 5], [6, 7, 8], [9]]
Sử dụng itertools.zip_longest
trong Python
Thư viện itertools
có thể giúp bạn chia danh sách với các phần tử cuối cùng được điền thêm None
nếu phần cuối không đủ số phần tử:
from itertools import zip_longest def chunk(lst, n): return zip_longest(*[iter(lst)]*n, fillvalue=None) chunks = list(chunk(my_list, 3)) print(chunks) # Kết quả: [(0, 1, 2), (3, 4, 5), (6, 7, 8), (9, None, None)]
Sử dụng itertools.islice
trong Python
Với itertools.islice
, phần cuối cùng sẽ không được điền thêm nếu thiếu phần tử:
from itertools import islice def chunk(lst, n): it = iter(lst) return iter(lambda: tuple(islice(it, n)), ()) chunks = list(chunk(my_list, 3)) print(chunks) # Kết quả: [(0, 1, 2), (3, 4, 5), (6, 7, 8), (9,)]
Sử dụng itertools.batched
(Mới trong Python 3.12)
Từ phiên bản Python 3.12, bạn có thể dùng hàm itertools.batched
:]
from itertools import batched chunks = list(batched(my_list, 3)) print(chunks) # Kết quả: [(0, 1, 2), (3, 4, 5), (6, 7, 8), (9,)]
Sử dụng more-itertools
trong Python
Nếu bạn không sử dụng Python 3.12 và muốn sử dụng các công cụ mạnh hơn, có thể cài đặt thư viện more-itertools
:
Cài đặt:
pip install more-itertools
Sau khi cài đặt, bạn có thể dùng các phương thức sau:
import more_itertools as mit chunks = list(mit.chunked(my_list, 3)) print(chunks) # Kết quả: [[0, 1, 2], [3, 4, 5], [6, 7, 8], [9]] chunks = list(mit.grouper(3, my_list)) print(chunks) # Kết quả:
Kết bài
Trên đây là các phương pháp khác nhau để chia một danh sách trong Python thành các phần có kích thước bằng nhau. Bạn có thể chọn phương pháp phù hợp tùy theo nhu cầu và phiên bản Python mà bạn đang sử dụng. Từ việc tự viết hàm sinh, sử dụng các hàm trong thư viện itertools
cho đến việc tận dụng các công cụ của more-itertools
, bạn có rất nhiều lựa chọn để thực hiện tác vụ này một cách hiệu quả.
[(0, 1, 2), (3, 4, 5), (6, 7, 8), (9, None, None)]