xlstools.py 文件源码

def write_sheet(data_mat, filename, sheetname='Sheet1', header=np.empty(0, dtype=object),
                header_format=None, data_format=np.empty(0, dtype=object)):
    wbk = xlwt.Workbook()
    sheet = wbk.add_sheet(sheetname)
    start_line = 0
    if len(header) == data_mat.shape[1]:
        for j in xrange(data_mat.shape[1]):
            if header_format is None:
                sheet.write(0, j, header[j])
            else:
                sheet.write(0, j, header[j], header_format)
        start_line = 1
    if data_format.size != 0 and len(data_format) != data_mat.shape[0]:
        raise Exception(
            'data_format should be the same length as rows of data_mat')
    for i in xrange(data_mat.shape[0]):
        for j in xrange(data_mat.shape[1]):
            if data_format.size == 0:
                sheet.write(i + start_line, j, data_mat[i][j])
            else:
                sheet.write(i + start_line, j, data_mat[i][j], data_format[i])
    wbk.save(filename)