r - 使用 ifelse 为 R 中的新数据框列赋值

Question

我有一个时间序列数据框，想创建一个新的数字列，其中的值是现有数字列的函数，并根据星期几列进行分配。
例如，我需要类似于以下代码的内容:

Day <- c("Mo", "Mo", "Mo", "Tu", "Tu", "We", "We", "We", "We", "Th")
Val <- c(1000, 1000, 1000, 1000, 1000, 1000, 1000, 1000, 1000, 1000)
df <- data.frame(cbind(Day,Val))
df$Adj <- ifelse(df$Day == "Mo" || df$Day == "Tu", 
                 as.numeric(levels(df$Val)) + 1, 
                 as.numeric(levels(df$Val)) + 2)

返回:

   Day  Val  Adj
1   Mo 1000 1001
2   Mo 1000 1001
3   Mo 1000 1001
4   Tu 1000 1001
5   Tu 1000 1001
6   We 1000 1002
7   We 1000 1002
8   We 1000 1002
9   We 1000 1002
10  Th 1000 1002

对我来说不幸的是，我的代码只返回 Adj 作为一列 1001。

   Day  Val  Adj
1   Mo 1000 1001
2   Mo 1000 1001
3   Mo 1000 1001
4   Tu 1000 1001
5   Tu 1000 1001
6   We 1000 1001
7   We 1000 1001
8   We 1000 1001
9   We 1000 1001
10  Th 1000 1001

我已经在“我们”行之一上测试了 ifelse，它确实有效......

> ifelse(df$Day[6] == "Mo" || df$Day[6] == "Tu", 
+        as.numeric(levels(df$Val[6])) + 1, 
+        as.numeric(levels(df$Val[6])) + 2)
[1] 1002

...但我似乎无法让它在整个列上工作，我已经理解这是 ifelse 函数相对于循环 if-else 语句的优势之一。
我的方法基于我能找到的最相似的问题
( Create new column in dataframe using if {} else {} in R ) 但没有快乐。我在这里缺少什么？

Answer 1

使用 dplyr:

Day <- c("Mo", "Mo", "Mo", "Tu", "Tu", "We", "We", "We", "We", "Th")
Val <- c(1000, 1000, 1000, 1000, 1000, 1000, 1000, 1000, 1000, 1000)
df <- data.frame(Day,Val)

library(dplyr)

df %>% rowwise %>% mutate(Adj = ifelse(Day == "Mo" || Day == "Tu", 
                           Val + 1, 
                           Val + 2))
#> # A tibble: 10 x 3
#>    Day     Val   Adj
#>    <chr> <dbl> <dbl>
#>  1 Mo     1000  1001
#>  2 Mo     1000  1001
#>  3 Mo     1000  1001
#>  4 Tu     1000  1001
#>  5 Tu     1000  1001
#>  6 We     1000  1002
#>  7 We     1000  1002
#>  8 We     1000  1002
#>  9 We     1000  1002
#> 10 Th     1000  1002

请注意 df <- data.frame(cbind(Day,Val))将 Val 转换为 character这可能不是你要找的。
您可以将其简化为 df <- data.frame(Day,Val)