Python Pandas Create Subset Dataframe Basedlabs