r - 将字符串转换为个位数并求和

Question

我已经尝试了几个小时来找到解决我认为是一项简单任务的方法，但我失败了。

我有一个由 3 个不同字符组成的字符串('I','R' & 'O')，长度从 1 到 6。
例如

IRRROO
RRORRR
IIR
RIRRO

每个字符代表一个数字I=1,R=2,O=3
，我需要将此字符串转换为单个数字，乘以位置并对结果求和。例如

IRRROO ---> (1*1)+(2*2)+(2*3)+(2*4)+(3*5)+(3*6) =52
IIR    ---> (1*1)+(1*2)+(2*3) =9

在此先感谢您的帮助。

Answer 1

因子具有数字等价物。您可以在此示例中很好地利用它：

# original
x1 <- "IRRROO"

         # 1    2    3
levs <- c("I", "R", "O")

# split the string and convert to factors, then to numeric
x1f <- as.numeric(factor(strsplit(x1, "")[[1]], levels=levs))

# tally it up 
sum(x1f * seq_along(x1f))

或者作为一个不错的单行函数：

sumValue <- function(x, levs=c("I", "R", "O")) 
    sum(seq.int(nchar(x)) *  as.numeric(factor(strsplit(x, "")[[1]], levels=levs)))

sumValue("IRRROO")
# [1] 52
sumValue("IIR")
# [1] 9

Answer 2

使用命名向量作为输入的替代版本，允许为每个字母分配任意数值：

vals <- c(I=1,R=2,O=3)
test <- c("IRRROO","RRORRR","IIR","RIRRO")

sumValue <- function(dat,vals) {
  tmp <- strsplit(dat,"")
  vapply(tmp,function(x) sum(seq_along(x) * vals[x]),numeric(1))
}

sumValue(test,vals)
[1] 52 45  9 33