-1

我正在尝试在数据框的行上应用一个函数，并根据列中每个元素的值返回一个值。我宁愿传递整个数据框而不是命名每个变量，因为实际代码有很多变量——这是一个简单的例子。

我已经尝试过purrr map_dbl，rowwise但都无法工作。请问有什么建议吗？

#sample df
df <- data.frame(Y=c("A","B","B","A","B"),
                  X=c(1,5,8,23,31))

#required result
Res <- data.frame(Y=c("A","B","B","A","B"),
                  X=c(1,5,8,23,31),
                  NewVal=c(10,500,800,230,3100)
                  )

#use mutate and map or rowwise etc
Res <- df %>%
  mutate(NewVal=map_dbl(.x=.,.f=FnAdd(.)))

Res <- df %>%
  rowwise() %>% 
  mutate(NewVal=FnAdd(.))


#sample fn
FnAdd <- function(Data){

  if(Data$Y=="A"){
    X=Data$X*10
  }  

  if(Data$Y=="B"){
    X=Data$X*100
  } 
  return(X)
}
4

2 回答 2

3

如果有多个值，最好有一个key/val数据集，加入然后做乘法

keyVal <- data.frame(Y = c("A", "B"), NewVal = c(10, 100))
df %>%
   left_join(keyVal) %>%
   mutate(NewVal = X*NewVal)
#  Y  X NewVal
#1 A  1     10
#2 B  5    500
#3 B  8    800
#4 A 23    230
#5 B 31   3100

目前尚不清楚实际数据集“Y”列中有多少唯一值。如果我们只有几个值，那么case_when可以使用

FnAdd <- function(Data){
   Data %>%
      mutate(NewVal = case_when(Y == "A" ~ X * 10,
                                Y == "B" ~ X *100,
                                TRUE ~ X)) 
}

FnAdd(df)
#   Y  X NewVal
#1 A  1     10
#2 B  5    500
#3 B  8    800
#4 A 23    230
#5 B 31   3100
于 2017-09-26T06:25:00.073 回答
1

您最初是在寻找使用dplyr's rowwise() 函数的解决方案，所以这里是该解决方案。这种方法的好处是您不需要创建单独的函数。

这是使用 if() 的版本

   df %>% 
   rowwise() %>% 
   mutate(NewVal = ifelse(Y == "A", X * 10,
                          ifelse(Y == "B", X * 100)))

这是使用 case_when 的版本：

df %>% 
   rowwise() %>% 
   mutate(NewVal = case_when(Y == "A" ~ X * 10,
                             Y == "B" ~ X * 100))
于 2017-09-28T21:10:07.350 回答