如何根据其他列中的模式从两列中的任意一列中选择值

Question

我有一个数据框如下：

       x<-c(1,8,2,4,3,1)
        Y<-c(2,3,4,1,7,6)
        z<-c("abs_ro","xyz_be","hdh_ro","ahh_ra","dhhd_te","hdj_ro")
        df<-data.frame(x,y,z)

我想创建一个新的列值，根据列z中的模式从两列x或y中选择一列。如果列z中的值以“_ro”结尾或“_be”代码从x中选择值，并且如果它以“_ra”或“_te”结束，则选择y的值。 X

`我尝试了许多方法，比如grepl，grep等，没有人工作过。

Answer 1

可以有各种方式来做到这一点。使用基数R的一种方法假设您总是拥有x或y列的模式之一。

df$value <- ifelse(grepl("_ro$|_be$", df$z), df$x, df$y)

df
#  x y       z value
#1 1 2  abs_ro     1
#2 8 3  xyz_be     8
#3 2 4  hdh_ro     2
#4 4 1  ahh_ra     1
#5 3 7 dhhd_te     7
#6 1 6  hdj_ro     1

---

或者如果可能有多种模式，我们也可以使用case_when

library(tidyverse)

df %>%
  mutate(value = case_when(str_detect(z, "_ro$|_be$") ~ x, 
                           str_detect(z, "_ra$|_te$") ~ y, 
                           TRUE ~ NA_integer_))

Answer 2

我们可以以矢量化的方式做到这一点，而不依赖于任何ifelse

df$value <- df[-3][cbind(seq_len(nrow(df)), (!grepl("_(ro|be)$", df$z)) + 1)]
df$value
#[1] 1 8 2 1 7 1