使用 Arcpy 批量处理光栅图像

Question

我正在尝试用 Python 编写代码，该代码将堆叠 5 个波段光栅图像，这些图像都按顺序列在一个文件夹中，然后将堆叠的图像输出到一个新文件夹。我的第一直觉是使用复合带工具在 arcpy 中自动执行某种 for 循环结构。

我需要以下方面的帮助：

1. 我在开始使用 for 循环时遇到问题。关于如何解决这个问题的任何建议？

   import arcpy
   arcpy.env.workspace = ".\\"
   outws = "Stacked_Images_Folder"
   
   for rasters in folder:
       band1 = 
       band2 = 
       band3 = 
       band4 = 
       band5 =
   
       arcpy.CompositeBands_management("band1.tif;band2.tif;band3.tif;
       band4.tif, band5.tif","stacked_img.tif")
   

2. 我试图弄清楚在堆叠 5 个波段后脚本如何知道移动到新图像。我是否需要在开始之前将图像分类到单独的文件夹中，或者是否有解决方法，例如代码知道在达到 5 个波段后移动到下一张图像吗？

Answer 1

如果您的栅格都在同一个文件夹中，则不需要 for 循环来执行此操作：

import arcpy
wd="Y:/" #have this as your directory where all rasters are located
arcpy.env.workspace = wd
raster_list=arcpy.ListRasters("", "tif")
arcpy.CompositeBands_management(raster_list,"stacked_img.tif") #will save output on the same folder specified above.

如果要将其保存到新的子目录：

import os
outws = wd+"Stacked_Images_Folder/"
os.makedirs(outws)
arcpy.CompositeBands_management(raster_list, outws + "stacked_img.tif")

现在，如果您有多个要合并到同一个文件夹中的具有共同起始文件名的栅格集，例如 img1-b1、img1-b2 等，您可以使用以下简单实现使整个过程工作：

import arcpy
image_names=["img" + str(s) for s in range(1,143)]
wd="Y:/" #have this as your directory where all rasters are located
arcpy.env.workspace = wd
for image_name in image_names:
    print image_name
    raster_list=arcpy.ListRasters(image_name+"-*", "tif")
    import os
    outws = wd+"Stacked_Images_Folder/"
    os.makedirs(outws)
    arcpy.CompositeBands_management(raster_list, outws + image_name+ "_stacked_img.tif")