Reconstruct numpy array using indexing [duplicate]

Question

I have a numpy array of 3*k elements, where k is an integer. For instance, for k=3 I have the array A. Below, x, y, and z are filled by drawing elements from A as shown in the following example:

import numpy as np

A = np.arange(9)
>> array([0, 1, 2, 3, 4, 5, 6, 7, 8])

x = A[0::3]
>> array([0, 3, 6])

y = A[1::3]
array([1, 4, 7])

z = A[2::3]
>> array([2, 5, 8])

Now, I want to use x, y, and z so as to reconstruct A. Is there any convenient way of doing this without using for loops?

Umang Gupta · Accepted Answer · 2018-06-24 19:19:35Z

1

You can do

np.array([x,y,z]).T.reshape(-1)

answered Jun 24, 2018 at 19:19

Umang Gupta

16.6k7 gold badges51 silver badges69 bronze badges

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

5 Comments

nullgeppetto Over a year ago

It works like a charm! Many thanks.

Umang Gupta Over a year ago

Feel free to accept the answer..if you like :)

nullgeppetto Over a year ago

Sure, but you answered too quickly... So you need to wait 5 more minutes :)

Umang Gupta Over a year ago

oh sorry! Didnt know that. One more thing, can you check for some other value of k (other than 3)

nullgeppetto Over a year ago

Yeah, it seems that it works for any integer k.

Collectives™ on Stack Overflow

Reconstruct numpy array using indexing [duplicate]

1 Answer 1

5 Comments

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

5 Comments

Linked

Related