Merge a list containing numpy and numbers to one numpy array

Question

I have the following list I want to turn into one numpy. What is the best and most effective way to do this?

[[array([1, 2, 3]), 1], [array([1, 2, 3]), 2], [array([1, 2, 3]), 4], [array([4, 4, 4]), 3]]

Expected result:

[[1, 2, 3, 1],
 [1, 2, 3, 2],
 [1, 2, 3, 4],
 [4, 4, 4, 3]]

meTchaikovsky · Accepted Answer · 2020-11-20 00:34:23Z

2

You can use this

test = [[np.array([1, 2, 3]), 1], 
        [np.array([1, 2, 3]), 2], 
        [np.array([1, 2, 3]), 4], 
        [np.array([4, 4, 4]), 3]]

np.apply_along_axis(lambda x:np.hstack((x[0],[x[1]])),axis=1,arr=test)

answered Nov 20, 2020 at 0:34

meTchaikovsky

7,6963 gold badges18 silver badges37 bronze badges

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

Comments

Chanran Kim · Accepted Answer · 2020-11-20 00:16:39Z

0

import numpy as np

arrays = [[np.array([1, 2, 3]), 1], [np.array([1, 2, 3]), 2], [np.array([1, 2, 3]), 4], [np.array([4, 4, 4]), 3]]

new_array = np.empty([0, 4])
for i in range(len(arrays)):
    new_array = np.vstack((new_array, np.append(arrays[i][0], arrays[i][1])))
    
print(new_array)

array([[1., 2., 3., 1.],
       [1., 2., 3., 2.],
       [1., 2., 3., 4.],
       [4., 4., 4., 3.]])

answered Nov 20, 2020 at 0:16

Chanran Kim

4493 silver badges8 bronze badges

Collectives™ on Stack Overflow

Merge a list containing numpy and numbers to one numpy array

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related