Question

我正在尝试构建路径名列表。到目前为止，我的代码是：

os.chdir(inputDir)

if Resursive is False:
    filePathList = [os.path.join(inputDir, f) for f in os.listdir(inputDir) if f.endswith('.tif')]

if Resursive is True:
    for root, dirs, files in os.walk(inputDir):
        for file in files:
            if file.endswith('.tif'):
                filePathList = (os.path.join(root, file))

很明显，这会导致问题，在Recursive is True情况下，filePathList每次都会被覆盖。在其他语言中，我会做类似filePathList[i] = (os.path.join(root, file))的操作，但是使用walk file和files并不是可用作索引值的数字。

在Recursive is True情况下处理的最佳方法是什么？

Answer 1

os.chdir(inputDir)

if Resursive is False:
    filePathList = [os.path.join(inputDir, f) for f in os.listdir(inputDir) if f.endswith('.tif')]

if Resursive is True:
    filePathList = []
    for root, dirs, files in os.walk(inputDir):
        for file in files:
            if file.endswith('.tif'):
                filePathList.append(os.path.join(root, file))

使用os.walk的递归列表

1 个答案: