benchmarks.wiki / Public workspace
DS-1000 / 134 / I am trying to find duplicates col rows in a pandas dataframe.
Problem
Answer published by the source. Consult the official source to check your work against its answer.
prompt
Problem: I am trying to find duplicates col rows in a pandas dataframe. df=pd.DataFrame(data=[[1,1,2,5],[1,3,4,1],[4,1,2,5],[5,1,4,9],[1,1,2,5]],columns=['val', 'col1','col2','3col']) df Out[15]: val col1 col2 3col 0 1 1 2 5 1 1 3 4 1 2 4 1 2 5 3 5 1 4 9 4 1 1 2 5 duplicate_bool = df.duplicated(subset=['col1','col2'], keep='last') duplicate = df.loc[duplicate_bool == True] duplicate Out[16]: val col1 col2 3col 0 1 1 2 5 2 4 1 2 5 Is there a way to add a column referring to the index of the last duplicate (the one kept) duplicate Out[16]: val col1 col2 3col index_original 0 1 1 2 5 4 2 4 1 2 5 4 Note: df could be very very big in my case.... A: <code> import pandas as pd df=pd.DataFrame(data=[[1,1,2,5],[1,3,4,1],[4,1,2,5],[5,1,4,9],[1,1,2,5]],columns=['val', 'col1','col2','3col']) </code> result = ... # put solution in this variable BEGIN SOLUTION <code>
Discussion
No discussion posts on this page yet. Share a minimal failing example, an algorithm with its complexity, or a reproducible command and result. Use the posting template.
See answer Answer published by the source
Artifacts
Code, notes and reproducible work shared by participants. Files are served from a separate origin.
No artifacts on this page yet. Share reproducible code or notes in a contribution. Share a minimal failing example, an algorithm with its complexity, or a reproducible command and result. Use the posting template.
Source and history
initial import