python - 使用os.walk读取多个输入文件，以及GetMessage进度更新消息

Question

我找不到这个问题的答案，所以我们开始吧。我正在编写一个将从 CSV 文件中读取数据的脚本，并且我希望自动执行代码以重复并通读一个文件夹中的多个 CSV 文件。我正在尝试使用 os.walk，但遇到错误。

问题:1.如何正确使用os.walk？2. 在输入 CSV 文件时插入进度更新的最佳方式是什么？

我需要执行 os.walk 的脚本的第一部分如下:

import sys
import os
import arcpy
import csv
from arcpy import env

## Set overwrite 
arcpy.env.overwriteOutput = True
print arcpy.env.overwriteOutput

## Set workspace 
arcpy.env.workspace = "C:\\Home\\directory\\"
workspace = "C:\\Home\\directory\\"
print workspace

for root, dirs, files in os.walk('C:\\Home\\directory\\CSV\\'):  
    print root, dirs, files
    ## Directory where I'm keeping the CSVs
    full_path = os.path.join(root, files)       

    rows = csv.DictReader(open(full_path, "rb"))
    if os.path.exists(outpath)==False:
         os.mkdir(outpath)

    for row in rows: # here begin reading through the CSV for the rest of the script

Answer 1

如果您想坚持使用 os.walk，作为 Jason 回答的替代方案，那么我通常会执行以下操作来查找 CSV 文件，然后使用它们

fileList = []
count = 0

directorypath = "C:\Home\directory\CSV"

for dirname, dirnames, filesnames in os.walk(directorypath):
    for filename in filenames:
       if filename.endswith(".csv"):
          fileList.append(filename)
          count = count + 1

这样你就可以只使用 fileList 来处理