How To Create A Sub Dataframe In Python