-2

你可能会从 Kaggle 中认出这一点。我有多个名为 Soil_Type1 的列，一直到 Soil_Type40。如果该土壤类型不存在，则值为 0，如果存在，则值为 1。每行只能存在 1 种土壤类型。

我想创建一个新列，如果 Soil_Type1 = 1 则取值为 S1，如果 Soil_Type2 = 1 则取值为 S2 等。我可以蛮力进行，即每次一行。有没有办法循环这个？

train_raw[,16:53 := lapply(.SD, as.character), .SDcols =16:53 ]

train_raw[,Soil_Type := "" ]
train_raw[Soil_Type1 == 1, Soil_Type := "S1"]
train_raw[Soil_Type2 == 1, Soil_Type := "S2"]
train_raw[Soil_Type3 == 1, Soil_Type := "S3"]
train_raw[Soil_Type4 == 1, Soil_Type := "S4"]

编辑：

抱歉，这是您所说的可重现示例的意思吗？

train_raw <- data.table(Soil_Type = "", 
                        Soil_Type1 = c(0,0,0,1), 
                        Soil_Type2 = c(0,0,1,0), 
                        Soil_Type3 = c(1,1,0,0))

train_raw[,Soil_Type := "" ]
4

1 回答 1

1

感谢David Arenburg的回答。

train_raw[, Soil_Type := "" ]
indx <- which(names(train_raw) == "Soil_Type")
cols <- paste0("Soil_Type", 1:4)
for(j in 1:length(cols))
   set(train_raw,which(train_raw[[cols[j]]] == 1L), 
       j = indx, value = paste0("S", j))
于 2015-10-15T13:14:33.850 回答