/Python/After groupby how to add values in two rows to a list

After groupby how to add values in two rows to a list

In [1]: df = pd.DataFrame( {'a':['A','A','B','B','B','C'], 'b':[1,2,5,5,4,6]})
        df

Out[1]: 
   a  b
0  A  1
1  A  2
2  B  5
3  B  5
4  B  4
5  C  6

In [2]: df.groupby('a')['b'].apply(list)
Out[2]: 
a
A       [1, 2]
B    [5, 5, 4]
C          [6]
Name: b, dtype: object

In [3]: df1 = df.groupby('a')['b'].apply(list).reset_index(name='new')
        df1
Out[3]: 
   a        new
0  A     [1, 2]
1  B  [5, 5, 4]
2  C        [6]

Source: stackoverflow.com

Similar codes

0 vote160 views1 answer
Numpy get index of nan
Python
0 vote158 views1 answer
Numpy second derivative of array
Python
12 votes172 views5 answers
Python UUID
Python
14 votes156 views3 answers
Exit python terminal
Python
31 votes158 views3 answers
Pandas order dataframe by column of other dataframe
Python
222 votes581 views6 answers
Python random numbers
Python
4 votes198 views2 answers
Django template for range
Python Django
56 votes147 views2 answers
Convert date string to date time string python
Python
0 vote140 views1 answer
Python post np.array object
Python
0 vote153 views1 answer
Longest Palindromic Substring using DP
Python