python遍历查找文件夹中符合要求的文件

python遍历查找文件夹中符合要求的文件

样例1：

#! /usr/bin/env python
# -*- coding: utf-8 -*-

# Name: searchfile.py
# Description: Traverse to find the required files in the folder
# Requirements: Spatial Analyst Extension

import os
import string

inputFilePath= "G:\\aaa\\shixun_data"
threefile = [ ]
key = "ND"
for file in os.listdir(inputFilePath):
    if os.path.splitext(file)[1] == '.tif':  # 查找.tif文件
        if string.find(file, key) != -1:  # 满足条件往下进行
            sourcefile = os.path.join(inputFilePath, file)  # 拼路径
            threefile.append(sourcefile)
print threefile

结果1：

['G:\\aaa\\shixun_data\\NDSI.tif', 'G:\\aaa\\shixun_data\\NDVI.tif', 'G:\\aaa\\shixun_data\\NDVI_Lindi.tif', 'G:\\aaa\\shixun_data\\NDWI.tif', 'G:\\aaa\\shixun_data\\NDWI_water.tif']

样例2：

#! /usr/bin/env python
# -*- coding: utf-8 -*-

# Name: searchfile.py
# Description: Traverse to find the required files in the folder
# Requirements: Spatial Analyst Extension

import os
import string

inputFilePath= "G:\\aaa\\shixun_data"
threefile = [ '']
key = "ND"
for file in os.listdir(inputFilePath):
    if os.path.splitext(file)[1] == '.tif':  # 查找文件夹下的.tif文件
        if string.find(file, key) != -1:  # 满足条件往下进行
            sourcefile = os.path.join(inputFilePath, file)  # 拼路径
            threefile.append(sourcefile)  
print threefile

结果2：

['', 'G:\\aaa\\shixun_data\\NDSI.tif', 'G:\\aaa\\shixun_data\\NDVI.tif', 'G:\\aaa\\shixun_data\\NDVI_Lindi.tif', 'G:\\aaa\\shixun_data\\NDWI.tif', 'G:\\aaa\\shixun_data\\NDWI_water.tif']