在R语言中,可以使用函数factor()
或as.factor()
将一个分类变量转换为虚拟变量(也称为哑变量)。以下是一个示例代码:
# 创建一个包含分类变量的数据框
data <- data.frame(category = c("A", "B", "A", "C", "B"))
# 将分类变量转换为虚拟变量
data$dummy <- as.factor(data$category)
# 将虚拟变量转换为虚拟变量
dummy_variables <- model.matrix(~ category, data = data)
# 将虚拟变量合并到原始数据框中
data <- cbind(data, dummy_variables)
# 查看转换后的数据框
print(data)
这段代码首先创建了一个包含分类变量的数据框data
,然后使用as.factor()
函数将category
列转换为虚拟变量,并将结果存储在新列dummy
中。接着,使用model.matrix()
函数将虚拟变量category
转换为虚拟变量矩阵dummy_variables
,最后将虚拟变量矩阵合并到原始数据框data
中。