Easy way to apply transformation from `pandas.get_dummies` to new data?

后端 未结 1 1289
轻奢々
轻奢々 2020-12-08 05:26

Suppose I have a data frame data with strings that I want converted to indicators. I use pandas.get_dummies(data) to convert this to a dataset that

1条回答
  •  失恋的感觉
    2020-12-08 05:34

    you can create the dummies from the single new observation, and then reindex this frames columns using the columns from the original indicator matrix:

    import pandas as pd
    df = pd.DataFrame({'cat':['a','b','c','d'],'val':[1,2,5,10]})
    df1 = pd.get_dummies(pd.DataFrame({'cat':['a'],'val':[1]}))
    dummies_frame = pd.get_dummies(df)
    df1.reindex(columns = dummies_frame.columns, fill_value=0)
    

    returns:

            val     cat_a   cat_b   cat_c   cat_d
      0     1       1       0       0       0
    

    0 讨论(0)
提交回复
热议问题