Numpy split array menjadi bongkahan berukuran sama dengan sisa

Question

Apakah ada fungsi numpy yang membagi array menjadi bongkahan berukuran m yang sama (tidak termasuk sisa yang berukuran kurang dari m). Saya telah melihat fungsi np.array_split tetapi itu tidak memungkinkan Anda membagi dengan menentukan ukuran potongan.

Contoh dari apa yang saya cari ada di bawah ini:

X = np.arange(10)
split (X, size = 3)
-> [ [0,1,2],[3,4,5],[6,7,8], [9] ]

split (X, size = 4)
-> [ [0,1,2,3],[4,5,6,7],[8,9]]

split (X, size = 5)
-> [ [0,1,2,3,4],[5,6,7,8,9]]

Answer 1

Inilah salah satu cara dengan np.split+ np.arange/range-

def split_given_size(a, size):
    return np.split(a, np.arange(size,len(a),size))

Contoh berjalan -

In [140]: X
Out[140]: array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])

In [141]: split_given_size(X,3)
Out[141]: [array([0, 1, 2]), array([3, 4, 5]), array([6, 7, 8]), array([9])]

In [143]: split_given_size(X,4)
Out[143]: [array([0, 1, 2, 3]), array([4, 5, 6, 7]), array([8, 9])]

In [144]: split_given_size(X,5)
Out[144]: [array([0, 1, 2, 3, 4]), array([5, 6, 7, 8, 9])]

Answer 2

lst = np.arange(10)
n = 3
[lst[i:i + n] for i in range(0, len(lst), n)]