python读写excel

1、读execl

# -*-coding:utf-8 -*-
import os
import xlrd
import re
import xlwt


def get_xlsx(filename):
    dirpath = os.path.join(os.path.dirname(os.path.abspath(__file__)), "send")

    return os.path.join(dirpath,filename)


def write_xlsx(k,v,headlist):
    workbook = xlwt.Workbook()
    c = len(headlist)
    r = len(v)
    try:
        worksheet = workbook.add_sheet("资产漏洞数据")
    except Exception as e:
        pass
    else:
        pass
    finally:
        for i in range(0,c):
            worksheet.write(0, i, headlist[i])
        row_r = 1
        for i in v:
            row_c = 0
            for d in i:
                worksheet.write(row_r,row_c,d)
                row_c += 1
            row_r += 1


        # for i in range(1,r):
        #     for d in range(1,c):
        #         print(v[i][d])
        #         # worksheet.write(i,c,v[i][d])
        filename = k + ".xls"
        workbook.save(filename)



def get_sheet_obj(sheet,filename):
    filepath = get_xlsx(filename)
    excel_obj = xlrd.open_workbook(filepath)
    for i in excel_obj.sheet_names():
        if i == sheet:
            sheet_index = excel_obj.sheet_names().index(i)
            sheet_obj = excel_obj.sheet_by_index(sheet_index)
            return sheet_obj


def get_sheet_rows(sheet_obj):
    return sheet_obj.nrows


def get_sheet_col(sheet_obj):
    return sheet_obj.ncols


def get_cell_data(sheet_obj,row,col):
    data = sheet_obj.cell_value(row, col)
    return data

　　

2、写excel，这里需要注意，就是要一次性的把数据写进去，然后save，如果多次sava的话，后面会把前面的数据覆盖掉

 

import xlwt
def write_xlsx(filename,sheetname,alldata):
    workbook = xlwt.Workbook()

    worksheet = workbook.add_sheet(sheetname)

    worksheet.write(r,c,alldata)
    filename = filename + ".xls"
    workbook.save(filename)