在 R 中,可以使用 model.matrix
函数将分类变量转换为虚拟变量(也称为 one-hot 编码)。
例如,假设你有一个数据框 df
,其中包含一列名为 fruit
的分类变量,其取值为 "apple"、"banana"、"orange"。你可以使用以下代码将这一列转换为虚拟变量:
df_dummy <- model.matrix(~ fruit, data = df)
这会在 df_dummy
中创建三列:fruitapple
、fruitbanana
和 fruitorange
。对于每个观察,只有一列的值为 1,其余列的值均为 0。
你还可以使用 dummy.data.frame
函数来实现相同的转换:
df_dummy <- dummy.data.frame(df, names = "fruit")
</