使用条件重命名向量

Question

有一个向量看起来像这样

v <- c(2,5,8,5,6,6,6,6,6,6,4,3,35,7)

有没有办法可以把这些关卡变成

A,A1,A1,A1,A1,A1,A1,A1,A1,A1,A1,A1,A2,A1

我知道我可以使用 if(v < 2){out <- 'A'}、if(v < 10){out <- 'A1'} 和 if(v > 10){out <- 'A2'} 等 if 命令

但这非常慢，因为我的向量有近一百万个数字。有没有办法真正快速地做到这一点？理想情况下，我可以设置一个变量，其中包含应替换 A 的所有数字、应替换 A1 的 var，等等。

Answer 1

你可以试试cut

as.character(cut(v, breaks=c(0,2,10,Inf),
                labels=c('A', 'A1', 'A2')))
#[1] "A"  "A1" "A1" "A1" "A1" "A1" "A1" "A1" "A1" "A1" "A1" "A1" "A2" "A1"

或者findInterval（这样会更快）

 c('A', 'A1', 'A2')[findInterval(v, c(0,2.1, 10, Inf))]

更新

如果要替换的值不遵循任何特定模式并且是根据您自己设定的规则决定的，则从“v”创建一个唯一值向量，并根据您需要替换的值命名它，并且使用[ 更快地替换整个向量。

unname(setNames( c('A', 'A1', 'A1', 'A1', 'A1', 'A1', 'A2', 'A1'),
   unique(v))[as.character(v)])
#[1] "A"  "A1" "A1" "A1" "A1" "A1" "A1" "A1" "A1" "A1" "A1" "A1" "A2" "A1"

更新2

根据 OP 对 @agstudy 帖子的 cmets，如果要替换的值存储在几个向量中，则可能的 data.table 解决方案如下。假设与“v”匹配的“q”向量中的对应值应替换为“A”，“w”中的对应值应替换为“A1”，其余值为“A2”。在这里，我在“v”列表上使用setDT 创建了一个data.table，并使用“A2”预设了一个向量“V2”。将“V1”列设置为键（setkey），加入“q”，将对应的“V2”值分配为“A”，同样为“w”。

q <- c(3,7,6,9,8)
w <- c(4,14)
library(data.table)
setkey(setDT(list(v, V2=rep('A2', length(v)))), 
            V1)[J(q), V2:='A'][J(w), V2:='A1']$V2
 #[1] "A2" "A"  "A1" "A2" "A2" "A"  "A"  "A"  "A"  "A"  "A"  "A"  "A"  "A2"

Answer 2

使用矢量化ifelse，这应该更快（至少对于给定的示例）：

ifelse(v < 2 ,'A',ifelse(v < 10, 'A1', 'A2'))
[1] "A1" "A1" "A1" "A1" "A1" "A1" "A1" "A1" "A1" "A1" "A1" "A1" "A2" "A1"

更一般地说（在 cmets 之后）：

ifelse(v %in% vA ,'A',ifelse(v %in% vA1, 'A1', 'A2'))

其中vA和vA1是一些向量，例如：

vA <- c(3,7,6,9,8)
vA1 <- c(4)