ML Util - sabirribas/sabirribas.github.io GitHub Wiki

def onehotenc(df, series):
    x = series.dropna().map(lambda value: '%s_%s' % (series.name, value))
    newcols = []
    for newcol in x.unique():
        df[newcol] = (x == newcol).astype(int)
        newcols.append(newcol)
    return newcols