python之将所有整数列更改为类别 python

如果我有以下数据:

import pandas as pd 
df = pd.DataFrame({'a' : [0, 1, 1, 1, 0], 'b' : [2,3,4,5,3], 'c': ['a','b','c','d','e']}) 
df 
 
    a   b   c 
0   0   2   a 
1   1   3   b 
2   1   4   c 
3   1   5   d 
4   0   3   e 

有没有一种简单的方法可以将所有整数列转换为类别？我能做到:

df['a'] = df['a'].astype('category') 

对于一列，但如何一次全部？在 R 中我会这样做:

df %>% mutate_if(is.integer, as.factor) 

请您参考如下方法:

你必须做的:

df.select_dtypes(include=['int64']).astype('category')