CODEWITHSUNDEEP
pythonarrayslistnumpy
lydias
lydias

Reputation: 841

Python: Numpy combine arrays into 2x1 lists

I'm hoping to combine two arrays

 A: ([1,2,5,8])
 B: ([4,6,7,9])

to 

 C: ([[1,4],
      [2,6],
      [5,7],
      [8,9]])

I have tried insert, append and concatenate, they only lump all elements together without giving the dimension in C.

I'm new to Python, any help will be appreciated.

Upvotes: 1

Views: 1092

Answers (2)

Waleed Iqbal
Waleed Iqbal

Reputation: 106

According to your initial approach, you only need to use zip, which returns an iterator of tuples, where the i-th tuple contains the i-th element from each of the argument sequences or iterables. 

import numpy

A = numpy.array([1,2,5,8])
B = numpy.array([4,6,7,9])

print(list(zip(A, B)))

It will print:

[(1, 4), (2, 6), (5, 7), (8, 9)]

Upvotes: 1

user3483203
user3483203

Reputation: 51155

Use numpy.column_stack:

Stack 1-D arrays as columns into a 2-D array

np.column_stack((A, B))

array([[1, 4],
       [2, 6],
       [5, 7],
       [8, 9]])

Upvotes: 1

Related Questions