不灭的焱

革命尚未成功,同志仍须努力

作者:php-note.com  发布于:2021-01-13 21:57  分类:Python/数据分析  编辑

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