尝试将分类值转换为标称值时出现错误 "invalid literal for int() with base 10: 'video'"

Question

我有一个 Dataframe，我想将 'video' 和 'photo' 作为对象更改为 1 和 2 作为 int。
我使用以下代码但出现错误：

invalid literal for int() with base 10: 'video'

你能帮帮我吗？

df = pd.read_csv('dataset.csv')    
df[['status_type']] = df[['status_type']].astype('int')

Answer 1

您可以使用：

pd.factorize()

将类别更改为数字。