Tìm tất cả các tệp CSV trong một thư mục bằng cách sử dụng Python

Question

Tôi làm cách nào để tìm tất cả các tệp trong thư mục có đuôi .csv trong python?

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) ] 

Chức năng find_csv_filenames() trả về một danh sách các tên tập tin như dây đàn, mà nằm trong thư mục path_to_dir với hậu tố nhất định (theo mặc định, ".csv").

Phụ Lục

Làm thế nào để in tên tập tin:

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

sử dụng Python OS module để tìm tập tin csv trong một thư mục.

các ví dụ đơn giản là ở đây:

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

tôi đã phải nhận csv tập tin đã được trong các thư mục con, do đó, bằng cách sử dụng phản hồi từ tchlpr tôi sửa đổi nó để làm việc tốt nhất cho trường hợp sử dụng của tôi:

import os 
import glob 

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