creating a new column based on other columns in R

Question

I have a data frame as:

pred

   V1 V2 V3
2   0  1  0
4   0  0  1
5   0  0  1
8   0  0  1
11  0  1  0
16  0  1  0
20  1  0  0
21  0  1  0
24  0  1  0
26  0  1  0
31  0  0  1
32  1  0  0
34  1  0  0

I want to create a 4th column that is equal to "Intermediate" if V1 is equal to 1, is labeled as "Long" if V2 is equal to 1, and "Short" if V3 is equal to 1.

Thank you

Try `mutate()` with `case_when()` from `dplyr`: https://stackoverflow.com/a/44931475/6105259 — Emman, Nov 23 '21 at 07:12

user2974951 · Accepted Answer · 2021-11-23T07:32:07.853

2

Base R solution. With df==1 we check which elements of the data frame are equal to 1, return the indices of the matches, and then match with a new vector with the desired names.

tmp=which(df==1,arr.ind=T)    
tmp=tmp[order(tmp[,"row"]),]
c("Intermediate","Long","Short")[tmp[,"col"]]

 [1] "Long"         "Short"        "Short"        "Short"        "Long"        
 [6] "Long"         "Intermediate" "Long"         "Long"         "Long"        
[11] "Short"        "Intermediate" "Intermediate"

edited Nov 23 '21 at 07:32

answered Nov 23 '21 at 07:18

user2974951

9,535
1
17
24

OP asked to create a new column, so need to add the output back to the table at the end... – Emman Nov 23 '21 at 07:26
@Emman Since it is trivial to assign this output to a new column I expect the OP to figure it out. – user2974951 Nov 23 '21 at 07:27

score 1 · Answer 2 · answered Nov 23 '21 at 13:01

Here is another base R approach. If df is your data.frame, you can convert it to a matrix, multiply by a sequence from 1 to number of columns, and create a factor with the desired labels (in the same order as the columns).

df$new <- factor(as.matrix(df) %*% (1:ncol(df)), 
                 labels = c("Intermediate", "Long", "Short"))

Output

R> df
   V1 V2 V3          new
1   0  1  0         Long
2   0  0  1        Short
3   0  0  1        Short
4   0  0  1        Short
5   0  1  0         Long
6   0  1  0         Long
7   1  0  0 Intermediate
8   0  1  0         Long
9   0  1  0         Long
10  0  1  0         Long
11  0  0  1        Short
12  1  0  0 Intermediate
13  1  0  0 Intermediate

score 1 · Answer 3 · answered Nov 23 '21 at 13:11

Maybe you can add new column using the code below

> c("Intermediate", "Long", "Short")[rowSums(col(df) * df)]
 [1] "Long"         "Short"        "Short"        "Short"        "Long"        
 [6] "Long"         "Intermediate" "Long"         "Long"         "Long"
[11] "Short"        "Intermediate" "Intermediate

creating a new column based on other columns in R

3 Answers3

Linked