Python: Create List from table

Question

I think the solution is very simple but I don't have any idea why I am so stupid today....

I have a dataframe like this

A B C
1 a red
1 a blue
1 c blue
1 c yellow
2 b red
2 d blue
3 a green
3 c yellow 
3 d blue
3 d yellow 
3 f red

And the result should be simple like this only a list or one column. So the former 1st column should be the headline with a line break before.

1
a [red, blue]
c [blue, yellow]

2
b [red]
d [blue]

3
a [green]
c [yellow]
d [blue, yellow]
f [red]

yes lol... but sometimes... it is probably the age😂😂🙈 — amarokWPcom
– amarokWPcom, Commented Mar 7, 2022 at 19:41

Flursch · Accepted Answer · 2022-03-07 18:52:58Z

1

Let df be your DataFrame you defined above. Use pandas.DataFrame.groupby to create groups/subgroups and loop over them:

groups = df.groupby('A')

for g in groups:
    print(g[0]) # "headline" of the following block 
    subgroups = g[1].groupby('B')
    for sg in subgroups:
        print(sg[1]['B'].values[0], sg[1]['C'].values)
    print('  ')

This yields:

1
a ['red' 'blue']
c ['blue' 'yellow']
  
2
b ['red']
d ['blue']
  
3
a ['green']
c ['yellow']
d ['blue' 'yellow']
f ['red']

edited Mar 7, 2022 at 18:52

answered Mar 7, 2022 at 18:36

Flursch

4933 silver badges4 bronze badges

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

Collectives™ on Stack Overflow

Python: Create List from table

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related