如何用R语言编写循环批量处理文件并自动计算角度值

问题背景 #

我有一个包含25个CSV文件的文件夹，文件名从smithii1.csv到smithii25.csv。目前只能手动导入单个文件（比如smithii1.csv），执行一系列计算得到对应的angle1值，现在需要通过R语言实现自动化循环处理，得到angle1到angle25的所有结果。

单轮手动处理的脚本如下：

# 读取数据
sm1 <- read.csv("smithii1.csv", header = TRUE)
# 排序以确保选中第一张图片对应的数据
sm1 <- sm1[order(sm1$number),]
# 设置中心点
a <- 1/2*sm1[1,]$imagewidth
b <- 1/2*sm1[1,]$imageheight
# 计算中心点到第一个追踪点的像素距离
d1 <- sqrt((sm1[1,]$x-a)^2+(sm1[1,]$y-b)^2)
# 计算像素转实际距离（米）的系数，公式：r=2*height*tanθ/imagewidth
r1 = 2*sm1[1,]$height*sm1[1,]$model/sm1[1,]$imagewidth
# 转换为实际距离
d1=d1*r1
# 计算角度
angle1 <- atan(d1/sm1[1,]$height)*180/pi

自动化解决方案 #

以下是实现批量处理的脚本（由@langtang提供）：

# 设置目标文件夹路径，请替换为你的实际路径
setwd(folderpath)
# 获取所有符合命名规则的CSV文件路径
paths = dir(pattern = "smith.*csv$")

# 定义单个文件的处理函数
f <- function(path) {
  sm1 <- read.csv(path, header = TRUE)
  sm1 <- sm1[order(sm1$number),]
  a <- 1/2*sm1[1,]$imagewidth
  b <- 1/2*sm1[1,]$imageheight
  d1 <- sqrt((sm1[1,]$x-a)^2+(sm1[1,]$y-b)^2)
  r1 = 2*sm1[1,]$height*sm1[1,]$model/sm1[1,]$imagewidth
  d1=d1*r1
  angle1 <- atan(d1/sm1[1,]$height)*180/pi
  # 返回包含文件名和对应角度的数据框
  return(data.frame(file = path, angle=angle1))
}

# 批量处理所有文件并合并结果
sm.angle <- data.frame(do.call(rbind, lapply(paths, f)))

关键步骤说明 #

• setwd(folderpath)：设置工作目录为存放CSV文件的文件夹，记得将folderpath替换为你的实际路径。
• dir(pattern = "smith.*csv$")：通过正则匹配获取所有以smith开头、.csv结尾的文件路径，确保只处理目标文件。
• 自定义函数f(path)：将单轮处理的逻辑封装成函数，输入文件路径，输出包含文件名和对应角度的数据框。
• lapply(paths, f)：遍历所有文件路径，对每个文件执行处理函数；do.call(rbind, ...)将所有结果合并成一个完整的数据框，最终sm.angle会包含所有文件的名称和对应的角度值。

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