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

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] == 'k0