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

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

Official source

initial import