Diffusion Models As Dataframe Python Column Dataframe Drop