Iterating over multidimensional numpy arrays

Question

I have a numpy array of dimension (a,b,c). And I want to slice it to over c'th dimension. For eg: A numpy array of shape (2,3,4), I want to iterate over 4 arrays of dimension (2,3).

So far I have been doing

for i in range(c):
    arr = A[::,i]

But this doesn't compute the right thing. How can I compute this?

I have all the dimensions a,b,c. I want to slice the numpy array to get four 2*3 arrays. — Raj
– Raj, Commented Mar 30, 2019 at 12:42

Véronique Lamontagne · Accepted Answer · 2019-03-30 12:54:43Z

1

Doesn't it also take a comma between the colons? Like so : arr = A[:, :, i]

answered Mar 30, 2019 at 12:54

Véronique Lamontagne

471 silver badge9 bronze badges

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

Comments

abdullah.cu · Accepted Answer · 2019-03-30 12:54:50Z

0

You are missing a comma. Your code should look like:

for i in range(c):
    arr = A[:,:,i]

BTW the code above is computing the right thing, but you are writing a statement that doesn't solve your problem :)

answered Mar 30, 2019 at 12:54

abdullah.cu

6846 silver badges11 bronze badges

1 Comment

Raj Over a year ago

Can you provide some links to understand these kind of syntax

Collectives™ on Stack Overflow

Iterating over multidimensional numpy arrays

2 Answers 2

Comments

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related