numpy split ndarray based on indices

Question

For a data set, x, of dimensions (n, m):

n = 50
m = 100
x = np.random.random((n,m))

I would like to split it into y subsets, so that for 10 subsets it would be based on the indices:

index_1 = [0, 10, 20, 30, 40]
index_2 = [1, 11, 21, 31, 41]
...
index_9 = [8, 18, 28, 38, 48]
index_10 = [9, 19, 29, 39, 49]

I know that np.array_split() or np.split() can be used to subset based on the wanted number of subsets, and i would like a similar output (list of np.ndarrays)

[x[i::10] for i in range(10)]

Mechanic Pig
– Mechanic Pig

2022-10-04 13:42:25 +00:00
Commented Oct 4, 2022 at 13:42 — Mechanic Pig
– Mechanic Pig, Commented Oct 4, 2022 at 13:42

abo · Accepted Answer · 2022-10-04 14:02:58Z

1

you can use this as a reference and modify to your needs.

import numpy as np
n = 50
m = 100
mylist = np.random.randint(n, size=m)
for i in range(0, len(mylist), 5):
    print(mylist[i:i+5])

answered Oct 4, 2022 at 14:02

abo

863 bronze badges

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

Collectives™ on Stack Overflow

numpy split ndarray based on indices

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related