#!/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