请优化这个代码library(terra) path1 <- dir('G:/ERA5hr/SVWL1/1/',pattern = '*.tif',full.names = T)#导入地址后批量合并 path2 <- raster('G:/ERA5hr/Fw/a/a.tif') path3 <- raster('G:/ERA5hr/Fw/b/b.tif') path4 <- raster('G:/ERA5hr/Fw/wr/wr.tif') #nc5 <- stack(paths) for (i in 1:length(path1)){ print(path1[i]) sw <- stack(path1[i]) a <- path2 b <- path3 wr <- path4 df_sw <- as.matrix(sw) file_data <- ifelse(df_sw<wr, 1, sqrt(1+a*(df_sw-wr)^b)) result <- sw result <- setValues(result, file_data) output_path <- paste('G:/ERA5hr/Fw/1/', substr(path1[i],19,32),'.tif',sep='') writeRaster(result,output_path, overwrite=TRUE,format = "GTiff") }
时间: 2024-03-22 15:42:30 浏览: 54
首先,我们可以使用`raster::stack()`函数来一次性加载所有的TIFF文件,而不是一个一个地导入和合并。其次,我们可以将`path2`、`path3`和`path4`这三个raster对象转换为矩阵,并在循环外进行计算,以节省时间。
以下是优化后的代码:
``` r
library(raster)
# 导入所有TIFF文件
path1 <- list.files('G:/ERA5hr/SVWL1/1/', pattern = '*.tif', full.names = TRUE)
stack1 <- stack(path1)
# 将需要用到的raster对象转换为矩阵
a <- as.matrix(raster('G:/ERA5hr/Fw/a/a.tif'))
b <- as.matrix(raster('G:/ERA5hr/Fw/b/b.tif'))
wr <- as.matrix(raster('G:/ERA5hr/Fw/wr/wr.tif'))
# 遍历每个TIFF文件
for (i in 1:length(path1)) {
# 加载当前TIFF文件
sw <- stack1[[i]]
# 将stack对象转换为矩阵
df_sw <- as.matrix(sw)
# 计算结果
file_data <- ifelse(df_sw < wr, 1, sqrt(1 + a * (df_sw - wr) ^ b))
# 将结果赋值给raster对象
result <- sw
result <- setValues(result, file_data)
# 输出结果到磁盘
output_path <- paste('G:/ERA5hr/Fw/1/', substr(path1[i], 19, 32), '.tif', sep = '')
writeRaster(result, output_path, overwrite = TRUE, format = "GTiff")
}
```
希望这个优化的代码能够提高程序的运行效率。
阅读全文