PHP笔记网

os.walk(file_dir) 返回参数说明：返回一个三元组 dirpath, dirnames, filenames（文件夹下有文件夹和文件都可以）

例如：返回指定文件夹下所有CSV文件

import csv
import os
import pandas as pd

# 提取文件夹下的地址+文件名，源文件设定排序规则
def file_name(file_dir):
    L = []
    for root, dirs, files in os.walk(file_dir):
        for file in files:
            if os.path.splitext(file)[1] == '.csv':
                L.append(os.path.join(root , file))
        return L
print(file_name('C:/Users/wenjianbao/Desktop'))

os.listdir(path) 返回参数说明：返回一个二元组 dirpath, filenames （文件夹下只有文件）

例如：返回指定文件夹下的所有文件

import os

path = r'C:\Users\Administrator\Desktop\file'
for filename in os.listdir(path):
    print(os.path.join(path,filename))