python读取文件列表，pandas操作excel

最新推荐文章于 2024-09-29 18:07:31 发布

一个好学生-叶纪杉

最新推荐文章于 2024-09-29 18:07:31 发布

阅读量2.5k

点赞数 1

CC 4.0 BY-SA版权

分类专栏：数据处理 python pandas

本文链接：https://blog.youkuaiyun.com/zjjoebloggs/article/details/84792885

本文介绍了如何使用Python的os模块遍历指定文件夹获取Excel文件列表，然后利用Pandas进行数据处理，包括根据条件新增列，并最终保存到新的Excel文件中。虽然处理速度较慢，但内存占用较低，特别是使用data.loc[]选择DataFrame列非常方便。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

python读取文件列表，pandas操作excel

目标

读取指定文件下所有excel文件
根据excel文件一列的内容分条件新增列
保存并输入新excel文件

思路

读取文件利用os.walk()
判断、新增列利用pandas
输入新excel文件利用pandas的ExcelWriter()

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Mon Dec  3 13:27:49 2018

@author: yejishan
"""

import pandas as pd
import os


# 读取并存储输入文件名称
def get_files(input_root):
    file_list = []
    for root, dirs, files in os.walk(input_file):
        for i in files:
            if i.split('.')[-1] == 'xls':
                file_list.append(i)

    return file_list


# code处理
def add_code(file):
    data = pd.read_excel(input_file + os.path.sep + file)
    data['code'] = None
    res = data['res']

    for i in range(len(res)):
        if res[i] == 'k01':
            data.loc[i, 'code'] = int(str(data['trigger'][i]) + '01')
        elif res[i] ==