R self_函数无法工作无法子集结束后的列

对您的功能的最小修改:

1. 将"NA"替换为NA.前者是字面字符串，后者是指R中缺失的值.
2. 将前ifelse()替换为if (...) {...} else {...}.否则，当grouprow为1时，cumsum_single_data[grouprow-1]会抛出错误.
3. 将第二个ifelse()移至for循环之外，因为它可以进行向量化判断.

allocate_data_m <- function(single_data, total_data) {
  cumsum_single_data <- rep(NA, length(single_data))
  for (grouprow in seq_len(length(single_data))) {
    cumsum_single_data[grouprow] <- if(grouprow == 1) {
      single_data[grouprow]
    } else {
      cumsum_single_data[grouprow-1] + single_data[grouprow] 
    }
  }
  out <- ifelse(cumsum_single_data < total_data, single_data, 0)
  out 
}

raw_df %>%
  group_by(cat) %>%
  mutate(amount_x_all = allocate_data_m(single_amount, total_amount)) %>%
  ungroup()

# # A tibble: 9 × 4
#   cat   single_amount total_amount amount_x_all
#   <chr>         <dbl>        <dbl>        <dbl>
# 1 a                 1           20            1
# 2 a                 3           20            3
# 3 a                 7           20            7
# 4 b                 2           10            2
# 5 b                 1           10            1
# 6 b                 4           10            4
# 7 b                 6           10            0
# 8 c                 1            9            1
# 9 c                 6            9            6

版本 2

使用列名而不是列载体.在这种情况下，当您在mutate()内调用allocate_data_m()时，您必须使用带双引号的列名.

allocate_data_m <- function(data, single_data, total_data) {
  if(is(data, "tbl")) data <- as.data.frame(data)
  cumsum_single_data <- rep(NA, nrow(data))
  for (grouprow in seq_len(nrow(data))) {
    cumsum_single_data[grouprow] <- if(grouprow == 1) {
      data[grouprow, single_data]
    } else {
      cumsum_single_data[grouprow-1] + data[grouprow, single_data]   
    }
  }
  out <- ifelse(cumsum_single_data < data[, total_data], data[, single_data], 0)
  out
}

raw_df %>%
  group_by(cat) %>%
  mutate(amount_x_all = allocate_data_m(cur_data(), "single_amount", "total_amount")) %>%
  ungroup()