combining array and list elemint by index

Question

I have an array (which comes from a kdtree):

array =  [[a b c d e]
          [a b c d e]
          [a b c d e]]

and a list :

lst = [1, 2, 3, 4, 5]

I want to do some list comprehension (using array and lst) that makes it look like this:

desired_result = [[a, b, c, d, e, 1]
                  [a, b, c, d, e, 2]
                  [a, b, c, d, e, 3]]

I am familiar with list comprehension just not familiar enough to know how to deal with this.

Can you share the data type of the `array`? – medium-dimensional Oct 18 '22 at 20:24 — medium-dimensional, Oct 18 '22 at 20:24
yeah they are all numbers – JaredCusick Oct 18 '22 at 20:32 — JaredCusick, Oct 18 '22 at 20:32

score 1 · Accepted Answer · answered Oct 18 '22 at 20:24

1

If you want a list comprehension:

result = [l+[x] for l,x in zip(array, lst)]

answered Oct 18 '22 at 20:24

mozway

score 0 · Answer 2 · answered Oct 18 '22 at 20:46

0

An alternative way to list comprehension if array is of the type numpy.ndarray:

result = numpy.c_[array, lst]

Please check this answer for more details.

answered Oct 18 '22 at 20:46

2 Answers2