Split large Excel files into chunks of n

split_excel.py

import argparse

import xlrd
import xlwt


def chunks(l, n):
    """Yield successive n-sized chunks from l."""
    for i in range(0, len(l), n):
        yield l[i:i + n]


def split_excel(infile, chunk_size=100):
    workbook = xlrd.open_workbook(infile)
    sheet = workbook.sheet_by_index(0)

    _chunks = list(chunks(range(0, sheet.nrows), chunk_size))

    for _chunk in _chunks:
        wb = xlwt.Workbook()
        ws = wb.add_sheet('Sheet 1')

        rows = [sheet.row_values(row, 0) for row in _chunk]

        for rowx, row_values in enumerate(rows):
            for colx, cell_value in enumerate(row_values):
                ws.write(rowx, colx, cell_value)

        wb.save('output_' + str(_chunks.index(_chunk)) + '.xls')


if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument('-i', '--input', help='Input file name', required=True)
    args = parser.parse_args()

    split_excel(args.input, chunk_size=100)