python 查找具有特定扩展名的目录中的所有文件

Posted 2021-05-09

import os
if data_path: # data_path is a given directory
    data_filenames = []
    for f in os.listdir(data_path): 
        if f.endswith('.csv'): # list all csv files in the directory
            data_filenames.append(f)
else:
    print('No directory chosen')