перечислять только файлы каталога и игнорировать подкаталоги с помощью python

Я хотел бы перечислить только файлы каталога и пропустить подкаталоги.

В текущем коде также перечислены файлы в подкаталогах, но это не то, что я хочу:


import os 

list_files = []    
for root, dirs, files in os.walk(input_folder):
        for filename in files:
            joined = os.path.join(input_folder, filename)
            list_files.append(joined)

Файлы в каталоге для обработки выглядят следующим образом:

/headfolder/subfolder/subfile.pdf     #ignore this directory
/headfolder/subfolder2/subfile.pdf    #ignore this directory
/headfolder/file.pdf                  #get this path 
/headfolder/file2.pdf                 #get this path  

желаемый результат:

['/headfolder/file.pdf','/headfolder/file2.pdf ']

person blah    schedule 24.03.2021    source источник


Ответы (1)


files=[i for i in os.listdir() if os.path.isfile(i)]

Редактировать:

для списка всех файлов в определенном каталоге

import os
from os import listdir
from os.path import isfile, join

input_folder = 'headfolder/'
list_files = []

files = [f for f in listdir(input_folder) if isfile(join(input_folder, f))]
for filename in files:
    joined = os.path.join(input_folder, filename)
    list_files.append(joined)

print(list_files)
person Sina kzlca    schedule 24.03.2021
comment
мне нужно, чтобы ввод был определенным каталогом, а не текущим каталогом. Если я делаю os.listdir(input_folder) , я получаю пустой список в качестве вывода. - person blah; 24.03.2021
comment
редакция как раз то, что я искал, спасибо! - person blah; 24.03.2021