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.

mozway · Accepted Answer · 2022-10-18 20:24:03Z

1

If you want a list comprehension:

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

answered Oct 18, 2022 at 20:24

mozway

267k13 gold badges55 silver badges106 bronze badges

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

Comments

medium-dimensional · Accepted Answer · 2022-10-18 20:46:00Z

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, 2022 at 20:46

2,29114 silver badges25 bronze badges