Combining 2 lists of lists in python

Question

I am newbie in python. I need to merge 2 lists of lists based on the first element in each list. Here are the lists:

first_list = [['aaa', 551, 10972],
              ['bbb', 552, 10872],
              ['ccc', 553, 11103],
              ['ddd', 554, 10912]]

second_list = [['aaa', 240], 
               ['bbb', 120], 
               ['ccc', 325], 
               ['ddd', 270]]

The code I have written:

join_list = []
for x in range(0, len(first_list)):
    temp = [first_list[x], second_list[x][1]]
    join_list.append(temp)

The following is the output:

join_list =
[[['aaa', 551, 10972], 240],
 [['bbb', 552, 10872], 120],
 [['ccc', 553, 11103], 325],
 [['ddd', 554, 10912], 270]]

I need to flatten the list of the list in each element in join_list, e.g the first element in join_list[0] = ['histogram', 551, 10972, 240]. Also, if the order of the second list is change and if there is an addition of a sublist in the first element, the 2 lists is still can be merged based on the first index in each element. So, it works like the VLOOKUP or HLOOKUP in Ms.Excel. How can we do this?

jpp · Accepted Answer · 2018-04-24 12:10:35Z

2

This is one way which does not assume a specific ordering of either list.

d = dict(second_list)

res = [i + [d[i[0]]] for i in first_list]

[['aaa', 551, 10972, 240],
 ['bbb', 552, 10872, 120],
 ['ccc', 553, 11103, 325],
 ['ddd', 554, 10912, 270]]

edited Apr 24, 2018 at 12:10

answered Apr 24, 2018 at 12:06

jpp

166k37 gold badges301 silver badges363 bronze badges

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

Comments

zipa · Accepted Answer · 2018-04-24 12:10:00Z

2

You could introduce lookup dict and set default value in case there is no match:

lookup = dict(second_list)
default = 0
join_list = [i + [lookup.get(i[0], default)] for i in first_list]

Here, default is set to 0.

answered Apr 24, 2018 at 12:10

zipa

28k6 gold badges45 silver badges62 bronze badges

Comments

Rakesh · Accepted Answer · 2018-04-24 12:13:08Z

1

Use chain from itertools to flatten

Demo:

first_list = [['aaa', 551, 10972],
              ['bbb', 552, 10872],
              ['ccc', 553, 11103],
              ['ddd', 554, 10912]]

second_list = [['aaa', 240], 
               ['bbb', 120], 
               ['ccc', 325], 
               ['ddd', 270]]

from itertools import chain
print([list(chain(*[v, [second_list[i][1]]])) for i, v in enumerate(first_list)])

Output:

[['aaa', 551, 10972, 240], ['bbb', 552, 10872, 120], ['ccc', 553, 11103, 325], ['ddd', 554, 10912, 270]]

edited Apr 24, 2018 at 12:13

answered Apr 24, 2018 at 12:05

Rakesh

82.9k17 gold badges85 silver badges122 bronze badges

3 Comments

YusufUMS Over a year ago

What's the difference between your output and mine? I need them to be flattened. And is it possible to merge them if the order of the second_list is changed?

Rakesh Over a year ago

Sorry mis-understood your que

Rakesh Over a year ago

You can use chain from itertools to flatten.

DSLima90 · Accepted Answer · 2018-04-24 12:14:32Z

1

Maybe this is what you are looking for:

d = dict(second_list)
flat = [a+[d[a[0]]] for a in first_list]

It gives me:

flat = [['aaa', 551, 10972, 240],
 ['bbb', 552, 10872, 120],
 ['ccc', 553, 11103, 325],
 ['ddd', 554, 10912, 270]]

This way the order of your second list does not matter.

answered Apr 24, 2018 at 12:14

DSLima90

2,8881 gold badge17 silver badges24 bronze badges

Collectives™ on Stack Overflow

Combining 2 lists of lists in python

4 Answers 4

Comments

Comments

3 Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

4 Answers 4

Comments

Comments

3 Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related