Pandas groupby on a column of lists

Question

I have a pandas dataframe with a column that contains lists:

df = pd.DataFrame({\'List\': [[\'once\', \'upon\'], [\'once\', \'upon\         


        

Answer 1

One way is to convert to tuples first. This is because pandas.groupby requires keys to be hashable. Tuples are immutable and hashable, but lists are not.

res = df.groupby(df['List'].map(tuple))['Count'].sum()

Result:

List
(a, time)       6
(once, upon)    5
(there, was)    1
Name: Count, dtype: int64

If you need the result as lists in a dataframe, you can convert back:

res = df.groupby(df['List'].map(tuple))['Count'].sum()
res['List'] = res['List'].map(list)

#            List  Count
# 0     [a, time]      6
# 1  [once, upon]      5
# 2  [there, was]      1