Merge numpy arrays contained in another array into different numpy array

Question

I'm working on numpy to manage different array. I have a python function that returns to me this array of arrays:

res = [[[1, 2], [5, 6], [7, 8]], [[1, 2], [5, 6], [7, 8]], [[1, 2], [5, 6], [7, 8]], [[1, 2], [5, 6], [7, 8]], [[1, 2], [5, 6], [7, 8]]]

What i desired to extract from this res variable is this one:

a = [1 2 1 2 1 2 1 2 1 2]
b = [5 6 5 6 5 6 5 6 5 6]
c = [7 8 7 8 7 8 7 8 7 8]

The idea is to extract and merge the first array of each array into a, the second arrays to b and so on.

Could you please help me to achieve this result? Thanks a lot!

Ch3steR · Accepted Answer · 2021-10-22 11:02:04Z

1

You can use numpy.hstack then use tuple unpacking.

In [7]: a, b, c = np.hstack(res)                                                

In [8]: a                                                                       
Out[8]: array([1, 2, 1, 2, 1, 2, 1, 2, 1, 2])

In [9]: b                                                                       
Out[9]: array([5, 6, 5, 6, 5, 6, 5, 6, 5, 6])

In [10]: c                                                                      
Out[10]: array([7, 8, 7, 8, 7, 8, 7, 8, 7, 8])

answered Oct 22, 2021 at 11:02

Ch3steR

20.8k4 gold badges34 silver badges66 bronze badges

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

1 Comment

Shreyas Prakash Over a year ago

Simple and effective!!!

Synthaze · Accepted Answer · 2021-10-22 11:02:13Z

0

Given:

import numpy as np

res = np.array([[[1, 2], [5, 6], [7, 8]], [[1, 2], [5, 6], [7, 8]], [[1, 2], [5, 6], [7, 8]], [[1, 2], [5, 6], [7, 8]], [[1, 2], [5, 6], [7, 8]]])

a = res[:, 0]
b = res[:, 1]
c = res[:, 2]

print(a)
print(b)
print(c)

Output:

[[1 2]
 [1 2]
 [1 2]
 [1 2]
 [1 2]]
[[5 6]
 [5 6]
 [5 6]
 [5 6]
 [5 6]]
[[7 8]
 [7 8]
 [7 8]
 [7 8]
 [7 8]]

Or:

print(a.flatten())
print(b.flatten())
print(c.flatten())

Output:

[1 2 1 2 1 2 1 2 1 2]
[5 6 5 6 5 6 5 6 5 6]
[7 8 7 8 7 8 7 8 7 8]

answered Oct 22, 2021 at 11:02

Synthaze

6,1082 gold badges16 silver badges35 bronze badges

Collectives™ on Stack Overflow

Merge numpy arrays contained in another array into different numpy array

2 Answers 2

1 Comment

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

1 Comment

Comments

Your Answer

Sign up or log in

Post as a guest

Related