Znajdź wszystkie pliki CSV w katalogu przy użyciu Pythona

Question

Jak znaleźć wszystkie pliki w katalogu z rozszerzeniem .csv w pythonie?

Answer 1

from os import listdir 

def find_csv_filenames(path_to_dir, suffix=".csv"): 
    filenames = listdir(path_to_dir) 
    return [ filename for filename in filenames if filename.endswith(suffix) ] 

Funkcja find_csv_filenames() zwraca listę nazw jako ciągi, które znajdują się w katalogu path_to_dir z podanym przyrostkiem (domyślnie „.csv”).

Uzupełnienie

Jak wydrukować nazwy plików:

filenames = find_csv_filenames("my/directory") 
for name in filenames: 
    print name 

Answer 2

import os 
import glob 

path = 'c:\\' 
extension = 'csv' 
os.chdir(path) 
result = [i for i in glob.glob('*.{}'.format(extension))] 
print(result) 

Answer 3

używanie Python OS moduł znaleźć plik CSV w katalogu.

prostym przykładem jest tutaj:

import os 

# This is the path where you want to search 
path = r'd:' 

# this is the extension you want to detect 
extension = '.csv' 

for root, dirs_list, files_list in os.walk(path): 
    for file_name in files_list: 
     if os.path.splitext(file_name)[-1] == extension: 
      file_name_path = os.path.join(root, file_name) 
      print file_name 
      print file_name_path # This is the full path of the filter file 

Answer 4

musiałem dostać csv pliki, które znajdowały się w podkatalogach, dlatego korzystając z odpowiedzi od tchlpr I zmodyfikowane go do pracy najlepsze dla mojego przypadku użycia:

import os 
import glob 

os.chdir('/path/to/main/dir') 
result = glob.glob('*/**.csv') 
print(result)