Def Generate_dummies для объекта в переменную цифры c в python

Question

У меня есть набор данных с 28 объектными переменными, и я хочу сделать One Hot Encoding, чтобы преобразовать их в цифры c. Я использую эту функцию, но она не работает

def generate_dummies(dataframe, variable_name):
    _dummy_dataset = pd.get_dummies(dataframe[variable_name], prefix=variable_name)
    dataframe = pd.concat([dataframe, _dummy_dataset], axis=1)
    dataframe.drop(variable_name, axis=1, inplace=True)
    return dataframe

for i in df.select_dtypes(include = object).'cate_columns':
    df = generate_dummies(df,i)

Я поместил все свои переменные объекта в список с именем cate_columns

Answer 1

get_dummies применяется только к столбцам, которые имеют dtype == object, поэтому нет необходимости go один за другим. Вы также можете ограничить нужные столбцы, установив параметр columns:

df = pd.get_dummies(df, columns = cate_columns)
df.drop(columsn = cate_columns, inplace=True)