Using Python Create Dataframe With Subset