Python -- xlrd,xlwt,xlutils 讀寫同一個Excel

 

最近開始學習python,想做做簡單的自動化測試，需要讀寫excel,然后就找到了xlrd來讀取Excel文件，使用xlwt來生成Excel文件（可以控制Excel中單元格的格式），需要注意的是，用xlrd讀取excel是不能對其進行操作的：xlrd.open_workbook()方法返回xlrd.Book類型，是只讀的，不能對其進行操作。而xlwt.Workbook()返回的xlwt.Workbook類型的save(filepath)方法可以保存excel文件。

 因此對於讀取和生成Excel文件都非常容易處理，但是對於已經存在的Excel文件進行修改就比較麻煩了。不過，還有一個xlutils（依賴於xlrd和xlwt）提供復制excel文件內容和修改文件的功能。其實際也只是在xlrd.Book和xlwt.Workbook之間建立了一個管道而已.

 

一、操作Excel 常用方法

 xlrd,xlwt,xlutils一套工具,xlrd是用來讀取excl的,xlwt是用來寫入excel的

            xlutils,引用了xlrd和xlwt來做一些如合並,過濾,修改文件的操作,這個很多人推薦使用,但有一個缺陷,就是,他的一個

           工作表sheet只能寫入65535行,多了就不能寫了,解決方法可以是,每65535行新建一個工作表sheet或者向后移動幾列,

          然后寫入,他的列最大值為256,所以最多一個sheet文件可以寫入:256 * 65535 個數據，還有就是只能讀xlsx文件，不能寫xlsx文件。

         對於xlsx文件的操作見  Python - openpyxl 讀寫操作Excel 介紹。

# -*- coding: utf-8 -*-
#導入xlwt模塊
import xlwt
# 創建一個Workbook對象，這就相當於創建了一個Excel文件
book = xlwt.Workbook(encoding='utf-8', style_compression=0)
'''
Workbook類初始化時有encoding和style_compression參數
encoding:設置字符編碼，一般要這樣設置：w = Workbook(encoding='utf-8')，就可以在excel中輸出中文了。
默認是ascii。當然要記得在文件頭部添加：
#!/usr/bin/env python
# -*- coding: utf-8 -*-
style_compression:表示是否壓縮，不常用。
'''
#創建一個sheet對象，一個sheet對象對應Excel文件中的一張表格。
# 在電腦桌面右鍵新建一個Excel文件，其中就包含sheet1，sheet2，sheet3三張表
sheet = book.add_sheet('test', cell_overwrite_ok=True)
# 其中的test是這張表的名字,cell_overwrite_ok，表示是否可以覆蓋單元格，其實是Worksheet實例化的一個參數，默認值是False
# 向表test中添加數據
sheet.write(0, 0, 'EnglishName')  # 其中的'0-行, 0-列'指定表中的單元，'EnglishName'是向該單元寫入的內容
sheet.write(1, 0, 'Marcovaldo')
txt1 = '中文名字'
sheet.write(0, 1, txt1.decode('utf-8'))  # 此處需要將中文字符串解碼成unicode碼，否則會報錯
txt2 = '馬可瓦多'
sheet.write(1, 1, txt2.decode('utf-8'))
 
# 最后，將以上操作保存到指定的Excel文件中
book.save(r'e:\test1.xls')  # 在字符串前加r，聲明為raw字符串，這樣就不會處理其中的轉義了。否則，可能會報錯

 

# -*- coding: utf-8 -*-
import xlrd
xlsfile = r"C:\Users\Administrator\Desktop\test\Account.xls"# 打開指定路徑中的xls文件
book = xlrd.open_workbook(xlsfile)#得到Excel文件的book對象，實例化對象
sheet0 = book.sheet_by_index(0) # 通過sheet索引獲得sheet對象
print "1、",sheet0
sheet_name = book.sheet_names()[0]# 獲得指定索引的sheet表名字
print "2、",sheet_name
sheet1 = book.sheet_by_name(sheet_name)# 通過sheet名字來獲取，當然如果知道sheet名字就可以直接指定
nrows = sheet0.nrows    # 獲取行總數
print "3、",nrows
#循環打印每一行的內容
for i in range(nrows):
    print sheet1.row_values(i)
ncols = sheet0.ncols    #獲取列總數
print "4、",ncols
row_data = sheet0.row_values(0)     # 獲得第1行的數據列表
print row_data
col_data = sheet0.col_values(0)     # 獲得第1列的數據列表
print "5、",col_data
# 通過坐標讀取表格中的數據
cell_value1 = sheet0.cell_value(0, 0)
print "6、",cell_value1
cell_value2 = sheet0.cell_value(0, 1)
print "7、",cell_value2

  

二、Excel操作實例

#-*-coding:utf-8-*-
# Time:2017/9/20 20:13
# Author:YangYangJun

import xlrd,xlwt,xlutils
import os
import time
from xlutils.copy import copy

'''
導入 excel操作的三個模塊
xlrd  讀取操作
xlwt  寫入操作
xlutils 復制、剪切、粘貼等操作

'''

#獲取當前文件路徑
currntPath = os.getcwd()
print currntPath
#D:\WorkSpace\Python\Study\Selenium\PyExcel

#獲取excel文件路徑
excelPath = os.path.join(os.getcwd(),'ExcelData')
print excelPath
#D:\WorkSpace\Python\Study\Selenium\PyExcel\ExcelData

#遍歷獲取文件名
excelPathDir = os.listdir(excelPath)
print excelPathDir
#['PyExcel.xlsx']
#excel與程序在同一目錄下，可以直接數據文件名。

#excelName = 'PyExcel.xlsx'

#如果不在一個目錄下，最好填寫絕對路徑

#excelName = os.path.join(excelPath,'PyExcel.xlsx')
excelName = os.path.join(excelPath,excelPathDir.pop())
print  excelName
#D:\WorkSpace\Python\Study\Selenium\PyExcel\ExcelData\PyExcel.xlsx

ReadExcel = xlrd.open_workbook(excelName)

print ReadExcel.sheet_names()
# 返回的是sheet頁list [u'userInfo', u'tests']

 

讀取excel根據條件修改對應表格內容，並保存。

原始表格：

 

 

 

代碼實例：

#-*-coding:utf-8-*-
# Time:2017/9/20 20:13
# Author:YangYangJun

import xlrd,xlwt,xlutils
import os
import time
#導入copy 方法方便文件寫操作。
from xlutils.copy import copy

'''
導入 excel操作的三個模塊
xlrd  讀取操作
xlwt  寫入操作
xlutils 復制、剪切、粘貼等操作

'''


def OpenXlxs():

    xlsxPath = os.path.join(os.getcwd(),'ExcelData')
    xlsxPathDir = os.listdir(xlsxPath)
    xlsxName = os.path.join(xlsxPath,xlsxPathDir.pop())
    print xlsxName
    readOpenXlsx = xlrd.open_workbook(xlsxName)

    readXlsxSheet = readOpenXlsx.sheet_by_name('userInfo')
    # copy管道作用
    writeOpenXlsx = copy(readOpenXlsx)

    print readXlsxSheet,writeOpenXlsx,xlsxName
    return readXlsxSheet,writeOpenXlsx,xlsxName

def readXlsx(readXlsxSheet, writeOpenXlsx, xlsxName):
    #獲取行數
    print "***"
    rowMax = readXlsxSheet.nrows
    print rowMax # 4
    #獲取第一行的值
    rows = readXlsxSheet.row_values(0)
    print rows

    #獲取列數
    colMax = readXlsxSheet.ncols
    print colMax   # 8

    for r in range(rowMax):
        if r == 0:
            continue
        else:
            rows = readXlsxSheet.row_values(r)
            print rows
            RunValue = readXlsxSheet.cell(r,6).value
            if RunValue == 'Y':
                writeXlsx(writeOpenXlsx,r,xlsxName)


def writeXlsx(writeOpenXlsx,row,xlsxName):
    # 前面已經通過 copy 方法獲取了writeOpenXlsx
    # 通過get_sheet()獲取的sheet有write()方法
    writeXlsxSheet = writeOpenXlsx.get_sheet(0)
    modifyTime = time.strftime('%Y-%m-%d')
    print modifyTime
    writeXlsxSheet.write(row,4,modifyTime)
    writeXlsxSheet.write(row,7,'Pass')
    writeOpenXlsx.save(xlsxName)


if __name__ == '__main__':
    readXlsxSheet, writeOpenXlsx, xlsxName = OpenXlxs()

    readXlsx(readXlsxSheet, writeOpenXlsx, xlsxName)

 

 運行代碼后文件

 

 打開時報如圖信息

 

此時修改一下文件后綴名即可打開，修改為xls即可。運行結果如圖，根據條件修改了相應行的值。

 

 

另外一種處理辦法就是用openpyxl 庫操作xlsx格式文件。

  參考：Python - openpyxl 讀寫操作Excel