如何使用 purrr 替换 R 中的修改循环？

Question

我有一个这样的数据框：

df <- data.frame(
  Group = c('A', 'B', 'C', 'A', 'B', 'C', 'A', 'B', 'C'),
  Value = c(12, 88, 54, 76, 23, 44, 60, 52, 18)
)

我想将每个组缩放到 100 的中位数，并用新值替换Value列，因此数据框如下所示：

df_desired <- data.frame(
  Group = c('A', 'B', 'C', 'A', 'B', 'C', 'A', 'B', 'C'),
  Value = c(20, 169.23, 122.73, 126.67, 44.23, 100, 100, 100, 40.91)
)

使用像这样的 scale_helper：

scale_helper <- function(x, value) x * value / median(x)

我可以用for循环来做到这一点，但如果可能的话，我想改用purrr 。 是否有使用purrr的直接方法，或者for循环是更好的方法？

Answer 1

Loop for 不是好方法，但我不明白你为什么要使用 purr。 我认为，这是一个很好的版本：

df %>% group_by(Group) %>% mutate(Value = scale_helper(Value, 100)) %>% as.data.frame()

或者你可以使用data.table。 像这样的东西：

as.data.table(df)[, lapply(.SD, scale_helper, 100), keyby = Group]