如何在R语言中创建虚拟变量

在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中。