拆分一列,获取拆分列的平均值,然后更新结果

时间:2019-01-25 21:59:50

标签: r data.table strsplit

我目前正在尝试将包含一定范围值(作为字符)的列分成两个数字列,以计算一行中有一个范围的平均值。然后,我想用更新后的结果替换旧列。实际上,有很多专栏要做。.我试图找到一种方法,但是对我来说似乎是挑战。

下面是我尝试过的代码..它不起作用..

test.val <- data.table(id = c(1, 2, 3), 
                       colA = c("100-150", "200", "300"), 
                       colB = c("15", "20-30", "10"))
test.A <- test.val[, lapply(.SD, function(x){strsplit(x, split = "-")}), .SDcols = c("colA", "colB")]
test.B[, lapply(.SD, mean), .SDcols = c("colA", "colB")]

最后,我想了解以下内容:

   id colA colB
1:  1  125   15
2:  2  200   25
3:  3  300   10

有人可以帮助我吗? 非常感谢。

2 个答案:

答案 0 :(得分:1)

这是一种tidyverse可能性:

test.val %>%
 gather(var, val, -id) %>%
 separate(val, c("val1", "val2"), sep = "-", convert = TRUE) %>%
 mutate(res = rowMeans(.[, 3:4], na.rm = TRUE)) %>%
 select(-val1, -val2) %>%
 spread(var, res)

  id colA colB
1  1  125   15
2  2  200   25
3  3  300   10

第一步,它将数据从宽格式转换为长格式。然后,它将值分成两列。最后,它计算行均值并将数据转换回原始格式。

考虑到您要分隔的每一列可能有两个以上的值:

test.val %>%
 gather(var, val, -id) %>%
 mutate(val = strsplit(val, "-")) %>%
 unnest(val) %>%
 group_by(id, var) %>%
 mutate(res = mean(as.numeric(val))) %>%
 distinct(res) %>%
 spread(var, res)

答案 1 :(得分:1)

另一个使用data.table

的选项
library(data.table)
cols <- c("colA", "colB")
for(j in cols) {
  tmp <- vapply(strsplit(test.val[[j]], "-"), 
                FUN = function(i) mean(as.numeric(i)), 
                FUN.VALUE = numeric(1))
  set(test.val, j = j, value = tmp)
}
test.val
#   id colA colB
#1:  1  125   15
#2:  2  200   25
#3:  3  300   10

给出一个向量

x <- c("100-150", "200", "300")

strsplit的结果是一个字符向量列表

strsplit(x, "-")
#[[1]]
#[1] "100" "150"

#[[2]]
#[1] "200"

#[[3]]
#[1] "300"

我们将其包装到vapply中,然后在将每个矢量转换为数字后计算每个元素的均值。

vapply(strsplit(x, "-"), function(x) mean(as.numeric(x)), numeric(1))
# [1] 125 200 300

我们使用此结果,使用cols的{​​{1}}函数替换data.table中指定的每一列。