R将几个虚拟变量列合并为1

时间:2014-10-15 00:01:48

标签: r r-caret

我正在尝试合并汽车数据框(插入符号包)中的多个列。列"敞篷车,轿跑车,掀背车,轿车,马车"有虚拟变量,我想创建一个名为type的列,其列名取决于车辆的类型。

library(caret)
data(cars)
head(cars)
colnames(cars)

以下是汽车数据框的列名:

 [1] "Price"       "Mileage"     "Cylinder"    "Doors"       "Cruise"        
 [6] "Sound"       "Leather"     "Buick"       "Cadillac"    "Chevy"     
[11] "Pontiac"     "Saab"        "Saturn"      "convertible" "coupe"   
[16] "hatchback"   "sedan"       "wagon" 

如何将最后5个虚拟变量列合并/合并为1与相应的车辆类型?

非常感谢任何见解或帮助!

2 个答案:

答案 0 :(得分:7)

带有max.col的矢量化解决方案:

cars$type <- names(cars[14:18])[max.col(cars[14:18])]

作为替代解决方案,您可以使用match

cars$type <- names(cars[14:18])[apply(cars[14:18], 1, match, x = 1)] 

答案 1 :(得分:6)

ind <- apply(cars[,14:18],1,function(x) which(as.logical(x)))
cars$type <- colnames(cars[,14:18])[ind]
head(cars)