df = pd.DataFrame(
{
"drive": [1,1,2,2,2,3,3,3,4,4,4,5,5,6,6,7,7],
"team": ['home','home','away','away','away','home','home','home','away',
'away','away','home','home','away','away','home','home'],
"home_comfy_lead": [0,0,0,0,0,0,0,1,0,0,0,1,1,0,0,1,1],
"home_drives": [1,1,0,0,0,2,2,2,0,0,0,3,3,0,0,4,4],
'home_drives_with_comfy_lead': [0,0,0,0,0,0,0,1,0,0,0,2,2,0,0,3,3]
})