Python遍历子目录查找文件对

Question

I have a deep subfolder structure like this:

a/b/file1.txt
a/b/file1.doc
a/b/file2.txt
a/b/file2.doc
a/c/file3.txt
a/c/file3.doc
a/c/d/file4.txt
a/c/d/file4.doc

I want to extract all the .txt and .doc file pairs - eg into a list of tuples - file names are identical, just file types differ.

Best I have come up with so far is the following which doesnt look very efficient:

files = []
for root, dirs, files in os.walk(path):
    for filename in files:
        if os.path.isdir(os.path.join(os.path.abspath("."), filename)):
            file_list = os.listdir(filename)
            file_list_copy = file_list.copy()
            #for each in file_list of type .txt
            # find .doc of same name in file_list_copy
            #add the 2 to tuple nd append to list

Answer 1

May not be the most efficient but works:

Using shell command to move the types to separate folders (ran for both txt and doc extensions to create 2 folders):

find /path-to-files-root/ -type f -name '*.txt' -exec mv -i {} /new-path-to-files/txt/ \;

Then I ran:

def get_all_files(path, pattern):
#see https://stackoverflow.com/questions/17282887/getting-files-with-same-name-irrespective-of-their-extension
    datafiles = []
    for root,dirs,files in os.walk(path):
        for file in fnmatch.filter(files, pattern):
            datafiles.append(file)
    return datafiles

txt_files = [f for f in os.listdir(txt_path) if isfile(join(txt_path, f))]
doc_files = [f for f in os.listdir(doc_path) if isfile(join(doc_path, f))]
for i, txt_file in enumerate(txt_files):
    filename = (os.path.splitext(txt_file)[0])
    doc_files = get_all_files(doc_path, '{0}.doc'.format(filename))
    if len(doc_files)== 1:
        doc_file = doc_files[0]
        #do something with txt_file and doc_file