Python Subset Dataframe Based Bodywork