Program Python untuk Membagi Daftar Menjadi Potongan Berukuran Merata

Contoh 1: Menggunakan hasil

def split(list_a, chunk_size):

    for i in range(0, len(list_a), chunk_size):
        yield list_a[i:i + chunk_size]

chunk_size = 2
my_list = [1,2,3,4,5,6,7,8,9]
print(list(split(my_list, chunk_size)))

Keluaran

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

Kode di atas membagi larik yang diberikan menjadi daftar yang lebih kecil masing-masing berukuran 2. Anda dapat melakukan hal yang sama menggunakan kompresi daftar seperti di bawah ini.

chunk_size = 2
list_chunked = [my_list[i:i + chunk_size] for i in range(0, len(my_list), chunk_size)]
print(list_chunked)

Keluaran

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

Contoh 2: Menggunakan numpy

import numpy as np

my_list = [1,2,3,4,5,6,7,8,9]
print(np.array_split(my_list, 5))

Keluaran

[array([1, 2]), array([3, 4]), array([5, 6]), array([7, 8]), array([9])]

Catatan: Anda perlu menginstal numpy di sistem Anda.