Python Subset Dataframe Based Bodyworks