Python Subset Dataframe Based Definition