Home > Net >  R's equivaltent of Python's np.select(conditions, choices)
R's equivaltent of Python's np.select(conditions, choices)

Time:01-24

I'm trying to create a new column in an R dataframe based on a set of conditions that are mutually exclusive. There is a clever way to achieve this on python using np.select(conditions, choices), instead of np.where (See this solved question). I've been looking for an equivalent on R that allows me to avoid writing a gigantic nested ifelse (which is the equivalent of np.where) without any success.

The amount of conditions that I have can change and I'm implementing a function for this. Therefore, and equivalent could be really helpful. Is there any option to do this? I'm new in R and come from python.

Thank you!

CodePudding user response:

Yes, you can use case_when in R:

library(dplyr)
mtcars%>%
  mutate(cyl2=case_when(cyl>7~"High",
                        cyl==6~"Medium",
                        T~"Low"))

    mpg cyl  disp  hp drat    wt  qsec vs am gear carb   cyl2
1  21.0   6 160.0 110 3.90 2.620 16.46  0  1    4    4 Medium
2  21.0   6 160.0 110 3.90 2.875 17.02  0  1    4    4 Medium
3  22.8   4 108.0  93 3.85 2.320 18.61  1  1    4    1    Low
4  21.4   6 258.0 110 3.08 3.215 19.44  1  0    3    1 Medium
5  18.7   8 360.0 175 3.15 3.440 17.02  0  0    3    2   High

CodePudding user response:

There's also cut(), Convert Numeric to Factor, with or without your own labels:

df <- data.frame(a = 1:10)

df$b <- cut(df$a, 
            breaks = c(-Inf,3,7,Inf), 
            labels = c("lo", "med", "hi"))

df$c <- cut(df$a, 
            breaks = c(-Inf,3,7,Inf))

df
#>     a   b        c
#> 1   1  lo (-Inf,3]
#> 2   2  lo (-Inf,3]
#> 3   3  lo (-Inf,3]
#> 4   4 med    (3,7]
#> 5   5 med    (3,7]
#> 6   6 med    (3,7]
#> 7   7 med    (3,7]
#> 8   8  hi (7, Inf]
#> 9   9  hi (7, Inf]
#> 10 10  hi (7, Inf]
  • Related