python 筛选包含需求字段表头的文件名称

python 筛选包含需求字段表头的文件名称

import os
import pandas as pd

# 需要筛选的文件夹,可以包含多个同类型的文件
file_path = 'C:/新建文件夹/'
os.chdir(file_path)
# 包含的字段名称
metric_fields = [‘name’,‘age’,‘garde’]
# 遍历当前文件夹下面所有的文件
for file in os.listdir():
    file_name = file.split('-')[1]
    csv_path = file_path + file
    df = pd.read_csv(csv_path, encoding='gbk')
    file_header_list = df.columns.to_list()
    for field in metric_fields:
        # 与表头字段进行对比,返回文件名称
        if field in file_header_list:
            print(file_name)

文件夹中包含的表格内容如下:
1、学生姓名表

idstudent
1张三
2李四
3王五

2、学生信息表

idnameage
1张三22

3、学生成绩表

idcoursegarde
1python100

程序运行结果:

学生信息表
学生成绩表

当我需要筛选表头包含有name,age,garde字段的文件时,就需要单个文件点开,去查看。如果文件很多,表头很多时,这会是一项费时费力的工作。通过本程序自动提取文件表头,识别需要的字段,返回你需要的文件名,就大大降低了工作量。
如有什么更好的,更优化的方法,欢迎交流。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值