如何使用 xlwt 和 xlrd 将每五行放入一个 sheet？

Question

现在我只能将13行数据全部保存到Sheet1。但实际上，我想把每5行放在一个sheet中，例如：

现在我的代码是这样的：

import xlwt
import xlrd
from xlutils.copy import copy
myyear = '2018'
myweek = '10'
mycd='1001'
fileName = myyear + myweek + '.xls'
workbook = xlwt.Workbook(encoding='UTF-8')
worksheet = workbook.add_sheet('sheet1')
worksheet.write(0, 0, label='year')
worksheet.write(0, 1, label='week')
worksheet.write(0, 2, label='CD')
worksheet.write(0, 3, label='keyword')
worksheet.write(0, 4, label='importance')
savePath = r'D:/' + fileName
workbook.save(savePath)
rexcel = xlrd.open_workbook(savePath)
rows = rexcel.sheets()[0].nrows
excel = copy(rexcel)
worksheet = excel.get_sheet(0)
current_row = rows
for cmp_noun, value in [[1,1],[2,2],[3,3],[4,4],[5,5],[6,6],[7,7],[8,8],[9,9],[10,10],[11,11],[12,12],[13,13]]:
    if current_row<=6:
        worksheet.write(current_row, 0, label=myyear)
        worksheet.write(current_row, 1, label=myweek)
        worksheet.write(current_row, 2, label=mycd)
        worksheet.write(current_row, 3, label=cmp_noun)
        worksheet.write(current_row, 4, label=value)
        current_row += 1
excel.save(savePath)

如何修改代码成为我想要的效果？

Answer 1

入门指南：

wb = Workbook()

for k in range(100):

    #create a new sheet 
    if not (k % 5):
        ws = wb.create_sheet('sheet {:d}'.format(k))

    # the rest is a bit of index magic using, indexing starts at 1 !
    ws.cell(row=1, column=1).value = 2

Answer 2

在逻辑段中构建代码，这样您就可以单独测试每个代码

虚拟数据

你的虚拟数据可以来自任何地方，这里我把它做成一个简单的生成器

def generate_data():
    year = '2018'
    week = '10'
    cd='1001'
    for i in range(1, 14):
        cmp_noun = value = i
        yield (year, week, cd, cmp_noun, value)

分组

from itertools import zip_longest
def grouper(iterable, n, fillvalue=None):
    """
    Collect data into fixed-length chunks or blocks
    https://docs.python.org/3.5/library/itertools.html#itertools-recipes
    """
    # grouper('ABCDEFG', 3, 'x') --> ABC DEF Gxx"
    args = [iter(iterable)] * n
    return zip_longest(*args, fillvalue=fillvalue)

list(grouper(generate_data(), 5))

对数据进行分组：

[(('2018', '10', '1001', 1, 1),
  ('2018', '10', '1001', 2, 2),
  ('2018', '10', '1001', 3, 3),
  ('2018', '10', '1001', 4, 4),
  ('2018', '10', '1001', 5, 5)),
 (('2018', '10', '1001', 6, 6),
  ('2018', '10', '1001', 7, 7),
  ('2018', '10', '1001', 8, 8),
  ('2018', '10', '1001', 9, 9),
  ('2018', '10', '1001', 10, 10)),
 (('2018', '10', '1001', 11, 11),
  ('2018', '10', '1001', 12, 12),
  ('2018', '10', '1001', 13, 13),
  None,
  None)]

制作练习册

def generate_workbook(header, data, n=5):
    groups = grouper(data, n)
    workbook = xlwt.Workbook(encoding='UTF-8')
    for i, group in enumerate(groups, 1):
        worksheet = workbook.add_sheet(f'sheet{i}')
        group = filter(None, group)
        populate_sheet(worksheet=worksheet, header=header, data=group)
    return workbook

这组数据，发起一个WorkBook。然后它为每个组制作一个新的工作表，并填充它。 filter(None, group)是当数据不是n的倍数时丢弃剩余的行。

填充工作表

def populate_sheet(worksheet, header, data):
    for i, column in enumerate(header):
        worksheet.write(0, i, column)
    for row_no, row in enumerate(data, 1):
        for i, item in enumerate(row):
            worksheet.write(row_no, i, label=item)

将 header 写入第一行，然后填充后续行。通过使用 enumerate 我们消除了很多 hard-coded 数字，并使整个数字更加通用

正在保存工作簿

if __name__ == '__main__':
    name = 'test.xls'
    data = generate_data()
    header = ['year', 'week', 'CD', 'keyword', 'importance']
    wb = generate_workbook(header=header, data=data, n = 5)
    wb.save(name)

如何使用 xlwt 和 xlrd 将每五行放入一个 sheet？

How to put every five rows into one sheet with xlwt and xlrd?

python

excel

xlrd

xlwt

虚拟数据

分组

制作练习册

填充工作表

正在保存工作簿