python实现Excel工作簿和表单合并

Created on Mon Mar 16 11:23:05 2015

@author: admin
"""
# 分批次运行，先得到books， 再运行，利用print 来debug

import pandas as pd
import numpy as np 
import matplotlib as pl
import os
count=0
#ori_path='xx'
ori_path='C:\Python27\data\sample'
def get_xls_books(ori_path): # get target filename and dir of a path 
    temp = list(os.walk(ori_path)) #tmp is list ,tmp[0]该目录文件路径和文件名， tmp[1]:子目录文件路径，文件名称 
    book_path=[]
    book_name=[]
    root=temp[0][0]
    for i in temp[0][2]:
      if os.path.splitext(i)[1] == '.xlsx': #splite filename 
          obj_path=os.path.join(root,i)
          book_path.append(obj_path)
          book_name.append(os.path.splitext(i)[0])
    return zip(book_path, book_name)

def get_xls_sheets(obj_path): #get all sheets of a object path
    xls=pd.ExcelFile(obj_path)  
    sheet_names=xls.sheet_names
    sheets=[xls.parse(i)  for i in range(0,len(sheet_names)) ]   # use parse can get dataframe saved into a list 
    return zip(sheet_names,sheets)

books=get_xls_books(ori_path)  # book paths list
#xls=pd.ExcelFile(books[0])
#print xls.parse(0).columns

for book in books:  # this loop get a book 
    bookname=book[1]
    bookpath=book[0]
    sheets=get_xls_sheets(bookpath)  # get all sheets and names from book path 
    for i in range(len(sheets)):  #operate each sheet, is range(len) not len 
        sheetname=sheets[i][0]
        name=bookname+'-'+sheetname+'.'+'csv' 
        pd_sheet=sheets[i][1] # dataFrame
        cols=pd_sheet.columns
        pdf=pd_sheet[[cols[17],cols[18],cols[4],cols[6],cols[8],cols[10],cols[19],cols[16]]]  # use emunerate function     
        pdf.columns=['area','patch','wangdian','operator','custid','servid','servtype','is_identify'] # rename 
        pdf1=pdf[pdf['is_identify']=='是']   #filter
        pdf2=pdf1.drop('is_identify',1)     
        pdf2.to_csv(os.path.join(ori_path,name), header=None, index=None)