Using Python Create Dataframe With Subset In R