Изменить тип данных столбца более питоническим способом

Question

Пожалуйста, помогите мне быть более питоническим:

Я помечаю все категориальные функции с помощью панд. Я знаю, что это также можно сделать со Sklearn, но я бы хотел сделать это только с Pandas или Python.

Я сделал это, сначала выбрав все столбцы типа 'obj', которые оказались cat (я имею дело с небольшим фреймом данных, поэтому я точно знаю это). Затем я использовал цикл for для преобразования каждого столбца.

Я знаю, что определенно могу сделать это, избегая цикла for. Чем больше питона, тем лучше:

cat_cols = df.select_dtypes(include='object').columns

for col in cat_cols:
    df[col] = df[col].astype('category').cat.codes

Answer 1

База на этой ссылке для цикла не всегда «плохая», если вам нужно ее использовать, вы можете использовать apply

cat_cols = df.select_dtypes(include='object').columns
df[cat_cols ] = df[cat_cols ].apply(lambda x : x.astype('category').cat.codes,axis=1)