I’ve a pd df consists three columns: ID
, t
, and ind1
.
import pandas as pd
dat = {'ID': [1,1,1,1,2,2,2,3,3,3,3,4,4,4,5,5,6,6,6],
't': [0,1,2,3,0,1,2,0,1,2,3,0,1,2,0,1,0,1,2],
'ind1' : [1,1,1,1,0,0,0,0,0,0,0,1,1,1,1,1,0,0,0]
}
df = pd.DataFrame(dat, columns = ['ID', 't', 'ind1'])
print (df)
What I need to do is to create a new column (res
) that
- for all
ID
withind1==0
, thenres
is zero. - for all
ID
withind1==1
and ift==max(t)
(group byID
), thenres
= 1, otherwise zero.
Here’s anticipated output
Aucun commentaire:
Enregistrer un commentaire