Computer >> Máy Tính >  >> Lập trình >> Python

Làm phẳng danh sách theo nhóm trong Python

Trong hướng dẫn này, chúng ta sẽ viết một chương trình làm phẳng một danh sách có chứa các danh sách con. Số đã cho san phẳng các danh sách con cho đến khi chỉ số số đã cho dưới dạng các bộ phận. Hãy xem một ví dụ để hiểu rõ điều đó.

Đầu vào

lists = [[1, 2], [3, 4], [5, 6], [7, 8], [9, 10]]
number = 2

Đầu ra

[[1, 2, 3, 4], [5, 6, 7, 8], [9, 10]]

Hãy xem các bước để giải quyết vấn đề.

  • Khởi tạo danh sách và số.
  • Khởi tạo một danh sách trống.
  • Lặp lại danh sách với phạm vi (0, len (danh sách), số.
  • Lấy danh sách phụ bằng cách sử dụng danh sách [i:number] cắt .
  • Lặp lại các danh sách phụ và nối danh sách kết quả vào danh sách kết quả.
  • In kết quả.

Ví dụ

# initializing the list
lists = [[1, 2], [3, 4], [5, 6], [7, 8], [9, 10]]
number = 2
# empty list
result = []
# iterating over the lists
for i in range(0, len(lists), number):
   # appending the lists until given number index each time
   result.append([element for sub_list in lists[i: i + number] for element in
   list])
   # printing the result
   print(result)

Đầu ra

Nếu bạn chạy đoạn mã trên, bạn sẽ nhận được kết quả sau.

[[1, 2, 3, 4], [5, 6, 7, 8], [9, 10]]

Kết luận

Nếu bạn có bất kỳ nghi ngờ nào trong hướng dẫn, hãy đề cập đến chúng trong phần bình luận.