R语言实现学年参与人数累加至缺失值（0标识）的技术求助

看起来你需要对每个学年的月份数据进行累加，从第一个非0的月份开始，直到遇到第一个代表缺失的0值为止（结合你的示例数据推断的逻辑，如果理解有误可以调整）。你之前的代码只针对Sep列处理，没有遍历所有月份列，也没实现停止条件，下面是具体的实现步骤：

1. 构造示例数据框 #

先把你提供的示例数据转换成可操作的R数据框：

yeardf <- data.frame(
  SchoolYear = c("1973-74", "1974-75"),
  Aug = c(0, 0),
  Sep = c(0, 0),
  Oct = c(4, 4),
  Nov = c(2, 26),
  Dec = c(14, 10),
  Jan = c(26, 23),
  Feb = c(22, 10),
  Mar = c(8, 14),
  Apr = c(0, 0),
  May = c(99, 91),
  stringsAsFactors = FALSE
)

2. 定义处理逻辑的函数 #

我们需要一个函数，对每个学年的月份向量执行以下操作：

• 跳过开头的所有0，找到第一个有数据的月份
• 从该月份开始累加，直到遇到第一个0就停止
• 同时可以计算累加过程中的最大值（对应你提到的Max_enroll）

calculate_enroll_stats <- function(month_vec) {
  # 找到第一个非0的月份位置
  first_non_zero <- which(month_vec != 0)[1]
  
  if (is.na(first_non_zero)) {
    # 如果所有月份都是0，返回0
    return(list(total = 0, max_enroll = 0))
  }
  
  # 截取从第一个非0到末尾的子向量
  target_vec <- month_vec[first_non_zero:length(month_vec)]
  # 找到子向量中第一个0的位置
  first_zero <- which(target_vec == 0)[1]
  
  if (is.na(first_zero)) {
    # 如果没有0，累加所有月份并计算最大值
    cumulative <- cumsum(target_vec)
  } else {
    # 累加第一个0之前的所有月份
    cumulative <- cumsum(target_vec[1:(first_zero - 1)])
  }
  
  # 返回累加总和和过程中的最大值
  list(Enroll_To_Date = tail(cumulative, 1), Max_enroll = max(cumulative))
}

3. 应用函数到数据框 #

使用apply按行处理月份列（排除SchoolYear列），然后把结果合并回原数据框：

# 按行处理月份数据
results <- apply(yeardf[, -1], 1, calculate_enroll_stats)

# 提取结果并添加到原数据框
yeardf$Enroll_To_Date <- sapply(results, function(x) x$Enroll_To_Date)
yeardf$Max_enroll <- sapply(results, function(x) x$Max_enroll)

4. 查看结果 #

运行后你的yeardf会变成：

> yeardf
  SchoolYear Aug Sep Oct Nov Dec Jan Feb Mar Apr May Enroll_To_Date Max_enroll
1    1973-74   0   0   4   2  14  26  22   8   0  99             76         76
2    1974-75   0   0   4  26  10  23  10  14   0  91             87         87

如果你的逻辑是从第一个月份开始累加，只要遇到0就停止（不管前面有没有非0），只需要修改函数里的逻辑，去掉找第一个非0的步骤，直接找整个向量的第一个0即可：

calculate_enroll_simple <- function(month_vec) {
  first_zero <- which(month_vec == 0)[1]
  if (is.na(first_zero)) {
    sum(month_vec)
  } else {
    sum(month_vec[1:(first_zero - 1)])
  }
}

这种情况下第一行的Enroll_To_Date会是0（因为第一个0在Aug），你可以根据实际需求选择。

内容的提问来源于stack exchange，提问作者regents