python - 使用 os.walk 获取 Python 中的文件路径

Question

我制作了一个简单的脚本来查找我的文件 test.txt，但我试图让它返回文件的位置。它只返回是否找到该文件，但我正在努力使该函数返回指定文件的文件路径，如果找到多个 test.txt，我希望它返回文件路径列表。

代码:

import os

wallet = 'test.txt'
filepath = 'C:\\'

def search():
    for root,dirs,files in os.walk(filepath):
        if wallet in files:
            return 'File found'
        else:
            return 'Nope.. not here'
print search() 

Answer 1

使用os.path.join将文件名与根连接起来并将其作为字符串返回:

import os

wallet = 'test.txt'
filepath = r'C:\\'

def search():
    for root,dirs,files in os.walk(filepath):
        if wallet in files:
            return os.path.join(root, wallet)
        else:
            return 'Nope.. not here'
print(search())

如果找到您的文件，应该打印:

C:\path_to_file\test.txt

PS:如果您将路径指定为'/'，我发现您正在使用Windows。它将具有相同的结果并提供工作路径，并且还具有与 Unix 兼容的优点。