How To Create Subset In Python