Data wrangling with python pandas

Question

I'm struggling to make do some data wrangling on a pandas dataframe. I've been stuck on this problem for 2 days now.

I've got a dataframe that looks like this:

['a','e']
['b','f']
['c','g']
['d','h']
['a','i']
['b','j']
['c','k']
['d','l']

And I need to turn it into looking like this

['a',['e','i']]
['b',['f','j']]
['c',['g','k']]
['d',['h','l']]

So basically pivoting the original dataframe around the first column and then creating a list of strings from the second column.

thanks

unutbu · Accepted Answer · 2015-03-03 03:08:53Z

3

You could use groupy/agg:

import pandas as pd

data = [['a','e'], ['b','f'], ['c','g'], ['d','h'], ['a','i'], ['b','j'], 
        ['c','k'], ['d','l']]

df = pd.DataFrame(data, columns=['first', 'second'])
print(df.groupby(['first']).agg(lambda x: x.tolist()))

yields

       second
first        
a      [e, i]
b      [f, j]
c      [g, k]
d      [h, l]

answered Mar 3, 2015 at 3:08

unutbu

886k197 gold badges1.9k silver badges1.7k bronze badges

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

Collectives™ on Stack Overflow

Data wrangling with python pandas

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