/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

5 votes149 views6 answers
Python speech to text
Python
0 vote153 views2 answers
Multiclasshead
Python
4 votes150 views3 answers
What is a character in python
Python
6 votes113 views1 answer
Read from yaml file python
Python
1 vote142 views1 answer
Cartesian product of a list python
Python
0 vote141 views1 answer
Import csv import re data = [] with open('customerData.csv') as csvfile: reader = csv.DictReader(csvfile) for row in reader: data.append(row) print(data[0]) print(data[1]["Name"]) print(data[2]["Spent Past 30 Days"])
Python
0 vote131 views1 answer
Finda argument index
Python
0 vote162 views1 answer
NameError: name 'CAP_PROP_FPS' is not defined
Python
0 vote139 views1 answer
Python program for simple interest
Python
10 votes147 views1 answer
How to varify pip is install in python
Python