Pandas - grouping IDs of each repeatable class and putting them in a list in column

Question

I have the dataframe like this:

d = {'repeat': [23, 25,23], 'unique': [1,2,3]}

df = pd.DataFrame(data=d)

Using the pandas built-in function the dataframe of this form:

grouped = {'unique': [[1,3],[2]], 'repeat': [23,25]}

dg =pd.DataFrame(data=grouped)

df -- pandas built-in operations --> dg?

Answer 1

You can use a groupby and agg like this:

df.groupby("repeat").agg(lambda x: list(x)).reset_index()

 Want to be a master in Data Science? Enroll in this Data Science Courses