如何使用for循环创建新列并提取文件以解答问题3和4？

API 探路者

2026-8-4

问题解答

问题背景

问题3和4要求针对不同窗口大小（3、4、6、8、10、14、20）计算径流的移动平均值，且每个窗口的计算结果需单独保存为文件。现有代码的核心问题是循环中反复覆盖同一列，无法保留所有窗口的计算结果。

数据示例

1.08E+03  -4.00E-06
1.08E+03  -5.60E-06
1.09E+03  -3.33E-06
1.09E+03  -4.60E-06
1.10E+03  -3.33E-06

修正后的代码

library(zoo)

# 读取数据（实际使用时可替换为read.table("你的数据文件路径")）
dat_10 <- read.table(text="1.08E+03  -4.00E-06
1.08E+03  -5.60E-06
1.09E+03  -3.33E-06
1.09E+03  -4.60E-06
1.10E+03  -3.33E-06", header = F)

# 修改列名，优化操作
colnames(dat_10) <- c("Time", "Runoff")

# 将Runoff列的0值替换为NA并移除含NA的行
dat_10$Runoff[dat_10$Runoff == 0] <- NA
dat_10 <- na.omit(dat_10)

# 将Runoff转为正数
dat_10$Runoff <- dat_10$Runoff * -1

# 定义需要计算的窗口大小
window_sizes <- c(3, 4, 6, 8, 10, 14, 20)

# 循环处理每个窗口
for (win in window_sizes) {
  # 生成唯一列名，避免覆盖数据
  col_name <- paste0("moving_avg_win", win)
  # 计算对应窗口的移动平均
  dat_10[[col_name]] <- rollmean(dat_10$Runoff, k = win, fill = NA)
  
  # 提取时间和当前窗口的移动平均结果
  result_df <- dat_10[, c("Time", col_name)]
  # 重命名列名
  colnames(result_df) <- c("Time in seconds", paste0("Moving average (window ", win, ")"))
  
  # 保存为带窗口编号的文件
  write.table(result_df, file = paste0("10_win", win, ".dat"), 
              row.names = FALSE, quote = FALSE)
}