Split array by input() if remaining list size < input(), split with remaining

Question

How do i split a numpy array by input(), however if there is not enough values to evenly split, just split with the remaining values

Example for clarity:

Arr = np.array([1, 2, 4, 7, 9])

if the user inputs 3 split the array by 3 but if there are not enough values to evenly split, just put the remaining values in one split

#Output
[[1, 2, 4], [7, 9]]

MaryRa · Accepted Answer · 2022-08-01 19:04:47Z

0

Maybe are you looking for something like this?

N = 3
np.array_split(Arr, range(N, len(Arr), N))

answered Aug 1, 2022 at 19:04

MaryRa

4632 silver badges4 bronze badges

Sign up to request clarification or add additional context in comments.

1 Answer 1