Pandas Create Dataframe Based On Another Dataframe In Databricks