在python中使用os.walk在子文件夹中找到文件的正确路径？

Question

I am trying to create a program that copies files with certain file extension to the given folder. When files are located in subfolders instead of the root folder the program fails to get correct path. In its current state the program works perfectly for the files in the root folder, but it crashes when it finds matching items in subfolders. The program tries to use rootfolder as directory instead of the correct subfolder.

My code is as follows

# Selective_copy.py walks through file tree and copies files with
# certain extension to give folder
import shutil
import os
import re

# Deciding the folders and extensions to be targeted
# TODO: user input instead of static values
extension = "zip"
source_folder = "/Users/viliheikkila/documents/kooditreeni/"
destination_folder = "/Users/viliheikkila/documents/test"

def Selective_copy(source_folder):
    # create regex to identify file extensions
    mo = re.compile(r"(\w+).(\w+)")  # Group(2) represents the file extension
    for dirpath, dirnames, filenames in os.walk(source_folder):
        for i in filenames:
            if mo.search(i).group(2) == extension:
                file_path = os.path.abspath(i)
                print("Copying from " + file_path + " to " + destination_folder)
                shutil.copy(file_path, destination_folder)

Selective_copy(source_folder)

Answer 1

dirpath是walk提供的其中一项功能，这是有原因的：它提供了files项目所在目录的路径。您可以使用它来确定您应该使用的子文件夹。

Answer 2

file_path = os.path.abspath(i)

This line is blatantly wrong. Keep in mind that filenames keeps list of base file names. At this point it's just a list of strings and (technically) they are not associated at all with files in filesystem.

os.path.abspath does string-only operations and attempts to merge file name with current working dir . As a result, merged filename points to file that does not exist .

What should be done is merge between root and base file name (both values yield from os.walk ):

file_path = os.path.abspath(dirpath, i)