Pandas dataframe issue

Question

I have this dictionary:

d = {1 : [ '2' , 'Pepsi' ], 3 : [ '1' , 'Pepper' ]}

df = pd.DataFrame([[slide, i[0], i[1]] for slide, item in d.items() for i in
item], columns=['Slide', 'Level', 'Level Title'])

I want to have the following output:

slide   level    'level1 title'   'level2 title'
1       2                          Pepsi
3       1         Pepper

So far, my code outputs this:

slide   level    'level title'
1       2         Pepsi
3       1         Pepper

Basically when the first item in the list is 1, the second item in the list should go to 'level1 title', and when the item is 2, the second item should go to 'level2 title'

Answer 1

If there are only two levels, then you can use list comprehension , like this -

In [12]: df = pd.DataFrame([[slide, item[0], item[1], ''] if item[0] == '1' else [slide, item[0], '', item[1]] 
for slide, item in d.items()], columns=['Slide', 'Level', 'Level1 Title', 'Level2 Title'])

In [13]: df
Out[13]:
   Slide Level Level1 Title Level2 Title
0      1     2                     Pepsi
1      3     1       Pepper