Python Script to iterate numerous, single-band rasters for arcpy.CompositeBands

CaseyCalamaio

I have a large collection of single-band geotiffs (NIR, Green, and Red) that I would like to composite using a Python script. The goal is to iterate through multiple folders in a main directory, identify each tiff as band1, band2, band3 to input into CompositeBands_management, and do this for every raster scene. To clarify, each image number, i.e. scene, has three single-band tiffs associated. For example, the naming convention is: TTC2500_Green_Channel_8.TIF where TTC2500 is the image number.

So far, my script can go through each folder and identify the specific bands based on the filename but I am having difficulty bringing them into the tool. The attached script has the Composite tool commented out but the result is a printed list of each geotiff. Any thoughts on how to move from here into an automated processing method to output new false color composite geotiffs?

import arcpy
import os
from arcpy.sa import *
from arcpy import env


env.workspace=(r"C:\Users\cc\Desktop\tcamPractice\point1_tif")
env.overwriteOutput = True


outws=(r"C:\Users\cc\Desktop\tcamPractice\false")
for root, dirs, files in os.walk(r"C:\Users\cc\Desktop\tcamPractice\point1_tif"):
    for raster in files:
        rasterlist=arcpy.ListRasters("*", "TIF")
        outraster=outws+"comp.tif"
        if raster.find("Green")>0:
            band1=raster
            print band1
        elif raster.find("Red")>0:
            band2=raster
            print band2
        elif raster.find("Nir")>0:
            band3=raster
            print band3
        arcpy.CompositeBands_management(raster, outraster)

jskinner

The below should work.  You can create a list of all the flights, i.e. TCC2500.  Then, remove the duplicates from this list.  Next, you can iterate through this list and create the composite based on the flight.

import arcpy
from arcpy import env
env.overwriteOutput = 1

workspace = r"C:\temp\python\UserData"

#create list
list = []

#populate list with flight, i.e. TTC2500
walk = arcpy.da.Walk(workspace, datatype="RasterDataset", type="TIF")
for dirpath, dirnames, filenames in walk:
    for raster in filenames:
        list.append(raster.split("_")[0])

#remove duplicates from list
list = dict.fromkeys(list)
list = list.keys()

for n in list:
    walk = arcpy.da.Walk(workspace, datatype="RasterDataset", type="TIF")
    for dirpath, dirnames, filenames in walk:
        env.workspace = dirpath
        for raster in filenames:
            if n + "_Green" in raster:
                green = env.workspace + "\\" + raster
            elif n + "_Red" in raster:
                red = env.workspace + "\\" + raster
            elif n + "_Nir" in raster:
                nir = env.workspace + "\\" + raster
        if green != '' and red != '' and nir != '':
            compBands = '"' + nir + ';' + green + ';' + red + ';"'
            compRasterName = "comp_" + str(n[3:]) + ".tif"
            outputRaster = env.workspace + "\\" + compRasterName
            arcpy.CompositeBands_management(compBands, outputRaster)