根据另一列中的4个值创建新列

时间:2011-10-05 07:58:34

标签: r if-statement

我想根据另一列中的4个值创建一个新列。

if col1=1 then col2= G;
if col1=2 then col2=H;
if col1=3 then col2=J;
if col1=4 then col2=K.

我如何在R中做到这一点? 我需要有人帮忙解决这个问题。我试过if / else和ifelse,但似乎没有工作。感谢

4 个答案:

答案 0 :(得分:26)

您可以使用嵌套的ifelse

col2 <- ifelse(col1==1, "G",
        ifelse(col1==2, "H",
        ifelse(col1==3, "J",
        ifelse(col1==4, "K",
                        NA  )))) # all other values map to NA

在这个简单的情况下,这是过度的,但对于更复杂的......

答案 1 :(得分:19)

您有一个特殊情况,即查找索引为整数1:4的值。这意味着您可以通过一个简单的步骤使用矢量索引来解决您的问题。

首先,创建一些示例数据:

set.seed(1)
dat <- data.frame(col1 = sample(1:4, 10, replace = TRUE))

接下来,定义查找值,并使用[子集来查找所需的结果:

values <- c("G", "H", "J", "K")
dat$col2 <- values[dat$col1]

结果:

dat
   col1 col2
1     2    H
2     2    H
3     3    J
4     4    K
5     1    G
6     4    K
7     4    K
8     3    J
9     3    J
10    1    G

更一般地说,您可以使用[子集与match结合来解决此类问题:

index <- c(1, 2, 3, 4)
values <- c("G", "H", "J", "K")
dat$col2 <- values[match(dat$col1, index)]
dat
   col1 col2
1     2    H
2     2    H
3     3    J
4     4    K
5     1    G
6     4    K
7     4    K
8     3    J
9     3    J
10    1    G

答案 2 :(得分:6)

有很多方法可以做到这一点,但这里有一个。

set.seed(357)
mydf <- data.frame(col1 = sample(1:4, 10, replace = TRUE))
mydf$col2 <- rep(NA, nrow(mydf))
mydf[mydf$col1 == 1, ][, "col2"] <- "A"
mydf[mydf$col1 == 2, ][, "col2"] <- "B"
mydf[mydf$col1 == 3, ][, "col2"] <- "C"
mydf[mydf$col1 == 4, ][, "col2"] <- "D"

   col1 col2
1     1    A
2     1    A
3     2    B
4     1    A
5     3    C
6     2    B
7     4    D
8     3    C
9     4    D
10    4    D

这是使用car的{​​{1}}。

recode

来自this question的另外一个:

library(car)
mydf$col3 <- recode(mydf$col1, "1" = 'A', "2" = 'B', "3" = 'C', "4" = 'D')

答案 3 :(得分:1)

您可以查看?symnum

在您的情况下,例如:

col2<-symnum(col1, seq(0.5, 4.5, by=1), symbols=c("G", "H", "J", "K"))

应该让你亲近。