Question

我有如下数据集：

df <- structure(list(city = structure(c(1L, 5L, 6L, 2L, 4L, 7L, 9L, 
8L, 3L), .Label = c("bj", "cq", "cs", "nj", "sh", "tj", "wh", 
"xa", "xm"), class = "factor"), area = c(1580.86, 1927.95, 532.24, 
613.09, 1459.86, 1024.58, 684.63, NA, 708.35), price = c(9.51, 
94.42, 10.77, 8.58, 38.07, 94.14, 30.43, 45.73, 41.39)), class = "data.frame", row.names = c(NA, 
-9L))

现在，我想根据以下条件创建一个新列level：

对于bj和sh，level的值为a；对于cq和tj，level的值为b；对于其他城市，值为c。

如何在R中做到这一点？谢谢。

Answer 1

使用dplyr::case_when和stringr::str_detect，您可以这样做：

library(dplyr)
library(stringr)

df %>% 
  mutate(level = case_when(str_detect(city, "bj|sh" ) ~ "a",
                           str_detect(city, "cq|tj" ) ~ "b",
                           TRUE ~ "c"))
#>   city    area price level
#> 1   bj 1580.86  9.51     a
#> 2   sh 1927.95 94.42     a
#> 3   tj  532.24 10.77     b
#> 4   cq  613.09  8.58     b
#> 5   nj 1459.86 38.07     c
#> 6   wh 1024.58 94.14     c
#> 7   xm  684.63 30.43     c
#> 8   xa      NA 45.73     c
#> 9   cs  708.35 41.39     c

^{由reprex package（v0.3.0）于2020-06-26创建}

Answer 2

library(tidyverse)
df %>% mutate(level = ifelse(city %in% c("bj", "sh"), "a", 
                             ifelse(city %in% c("cq", "tj"), "b", "c")))


  city    area price level
1   bj 1580.86  9.51     a
2   sh 1927.95 94.42     a
3   tj  532.24 10.77     b
4   cq  613.09  8.58     b
5   nj 1459.86 38.07     c
6   wh 1024.58 94.14     c
7   xm  684.63 30.43     c
8   xa      NA 45.73     c
9   cs  708.35 41.39     c

根据R中一个字符串列的多个条件创建一个新列

2 个答案: