Keep Other Variables When Executing Get_Dummies in Pandas
I Have a Dataframe with an Id Variable and Another Categorical Variable. I Want to Create Dummy Variables out of the Categorical Variable with Get_Dummies. Dum...
I have a DataFrame with an ID variable and another categorical variable. I want to create dummy variables out of the categorical variable with get_dummies.
dum = pd.get_dummies(df)
However, this makes the ID variable disappear. And I need this ID variable later on to merge to other data sets.
Is there a way to keep other variables. In the documentation of get_dummies I could not find anything. Thanks!
2 Answers
You can also copy the original column into a new one before executing get_dummies. E.g.,
df['dum_orig'] = df['dum']
df = pd.get_dummies(df, columns=['dum'])
I found the answer. You can concatenate the dummies data set to the original data set like shown below. As long as you don't re-order the data in the meantime.
df = pd.concat([df, dum], axis=1)
Sophia analyzes international trade, startup ecosystems, retail transformation, and supply chain logistics for modern digital publications.
