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

998
2024/5/22 12:16:07
栏目: 云计算
开发者测试专用服务器限时活动,0元免费领,库存有限,领完即止! 点击查看>>

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

辰迅云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>

推荐阅读: r语言怎么替换数据框的值