python爬虫学习Python 运维我的Python自学之路

python-读写excel

2016-06-13  本文已影响223人  望月成三人
import  sys
import xlrd
#https://pypi.python.org/pypi/xlrd/0.9.3
import xlsxwriter
#https://pypi.python.org/pypi/XlsxWriter/0.6.6#downloads
def read_excel(file='c:/test.xls'):
    data = xlrd.open_workbook(file)
    table = data.sheet_by_index(0)
    nrows = table.nrows
    ncols = table.ncols
    colnames = table.row_values(0) #one rows data
    list = []
    for rownum in range(1, nrows):
        row = table.row_values(rownum)
        if row:
            app = {}
            for i in range(len(colnames)):
                row[i] = colnames[i] + row[i]
                app[colnames[i]] = row[i]
            list.append(app)
    return list

def write_excel(file='d:/result.xlsx', httpurl=["",""], httpmethod=get, response_time=[0,10,14,0]):
    workbook = xlsxwriter.Workbook(file)
    worksheet = workbook.add_worksheet()
    #worksheet.set_column('A:A', 20)
    worksheet.write(0, 0, "接口URL")
    worksheet.write(0, 1, "请求方法")
    worksheet.write(0, 2, "响应时间")
    for i in range(len(response_time)):
        worksheet.write(i + 1, 0, httpurl[i])
        worksheet.write(i + 1, 1, httpmethod)
        if response_time[i] == 0:
            worksheet.write(i + 1, 2, "请求超时")
        else:
             worksheet.write(i + 1, 2, response_time[i])
    #worksheet.insert_image('B5', 'logo.png')
    workbook.close()

上一篇下一篇

猜你喜欢

热点阅读