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