how to limit the duplicate to 5 in pandas data frames?

Question

col1= ['A','B','A','C','A','B','A','C','A','C','A','A','A']
col2= [1,1,4,2,4,5,6,3,1,5,2,1,1]

df = pd.DataFrame({'col1':col1, 'col2':col2})

for A we have [1,4,4,6,1,2,1,1], 8 items but i want to limit the size to 5 while converting Data frame to dict/list

Output:

Dict = {'A':[1,4,4,6,1],'B':[1,5],'C':[2,3,5]}

sorry because of some error i did not put the original df. but now its updated. — Sunil
– Sunil, Commented Aug 20, 2019 at 6:31

Chris · Accepted Answer · 2019-08-20 06:34:21Z

7

Use pandas.DataFrame.groupby with apply:

df.groupby('col1')['col2'].apply(lambda x:list(x.head(5))).to_dict()

Output:

{'A': [1, 4, 4, 6, 1], 'B': [1, 5], 'C': [2, 3, 5]}

answered Aug 20, 2019 at 6:34

Chris

29.8k3 gold badges34 silver badges56 bronze badges

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

Comments

jezrael · Accepted Answer · 2019-08-20 06:33:57Z

4

Use DataFrame.groupby with lambda function, convert to list and filter first 5 values by indexing, last convert to dictionary by Series.to_dict:

d = df.groupby('col1')['col2'].apply(lambda x: x.tolist()[:5]).to_dict()
print (d)
{'A': [1, 4, 4, 6, 1], 'B': [1, 5], 'C': [2, 3, 5]}

answered Aug 20, 2019 at 6:33

jezrael

869k103 gold badges1.4k silver badges1.3k bronze badges