Оператор ifelse с использованием логического условия в R

Question

У меня есть этот фрейм данных

DF
ID    WORD_LIST   
1     APPLE
2     RED
3     SNOW
4     ANKARA
5     RENEW

Я хотел бы выбрать только слово, начинающееся с «A» или «R»

DF
ID    WORD_LIST    WORDS_A_R  
1     APPLE        APPLE
2     RED          RED
3     SNOW         NA
4     ANKARA       ANKARA
5     RENEW        RENEW

Я пробовал этот код

DF %>% mutate(WORDS_A_R = ifelse(grepl("^A" | "^R", WORD_LIST), as.character(WORD_LIST), NA))

Но возникает эта ошибка

operations are possible only for numeric, logical or complex types

Answer 1

Вы можете использовать:

transform(df, WORDS_A_R  = ifelse(grepl("^[AR]", WORD_LIST), WORD_LIST, NA))

Если хотите, dplyr:

library(dplyr)
df %>%
  mutate(WORDS_A_R  = ifelse(grepl("^[AR]", WORD_LIST), WORD_LIST, NA))

#  ID WORD_LIST WORDS_A_R
#1  1     APPLE     APPLE
#2  2       RED       RED
#3  3      SNOW      <NA>
#4  4    ANKARA    ANKARA
#5  5     RENEW     RENEW

Answer 2

Вы также можете использовать base R с функцией substring():

#Create condition
cond1 <- substr(DF$WORD_LIST,1,1)=='A' | substr(DF$WORD_LIST,1,1)=='R'
#Create var
DF$WORDS_A_R <- NA
DF$WORDS_A_R[cond1]<-DF$WORD_LIST[cond1]

ID WORD_LIST WORDS_A_R
1  1     APPLE     APPLE
2  2       RED       RED
3  3      SNOW      <NA>
4  4    ANKARA    ANKARA
5  5     RENEW     RENEW