Abhishek Kulkarni
Abhishek Kulkarni

Reputation: 3828

How to split merged cells in excel workbook using python

Is there any way to split/unmerge cells in excel workbook using python? What I want is explained below -

enter image description here

The result should a new excel file with following entries -

enter image description here

My solution using xlrd to copy the same string for all merged column is as given below -

[Note: "formatted_info = True" flag is not yet implemented in xlrd which I am using hence I cannot directly get the list of merged cells.. I am not supposed to upgrade xlrd on the setup.]

def xlsx_to_dict():
    workbook = xlrd.open_workbook(xlsfile)
    worksheet_names = workbook.sheet_names()
    for worksheet_name in worksheet_names:
        worksheet = workbook.sheet_by_name(worksheet_name)
        num_rows = worksheet.nrows - 1
        num_cells = worksheet.ncols - 1
        curr_row = -1
        header_row = worksheet.row(0)
        columns = []
        for cell in range(len(header_row)):
            value = worksheet.cell_value(0, cell)
            columns.append(value)

        cities = []

        for row in range(1,num_rows):
            value = worksheet.cell_value(row,0)
            type = worksheet.cell_type(row,0)
            if  not value == "":
                cities.append(value)

        names = []
        for row in range(1,num_rows):
            value = worksheet.cell_value(row,1)
            type = worksheet.cell_type(row,1)
            if  not value == "":
                names.append(value)

            current_city = cities[0]
            result_dict = {}
            for curr_row in range(1,num_rows):
                row = worksheet.row(curr_row)
                curr_cell = -1
                curr_name = names[0]
                while curr_cell < num_cells:
                    curr_cell += 1
                    cell_value = worksheet.cell_value(curr_row, curr_cell)
                    if cell_value in cities and curr_cell == 0:
                        current_city = cell_value
                        if not result_dict.has_key(current_city):
                            result_dict[current_city] = {}
                        continue
                    if cell_value == "" and curr_cell == 0:
                        continue
                    if cell_value in names and curr_cell == 1:
                        curr_name = cell_value
                        if not result_dict[current_city].has_key(curr_name):
                            result_dict[current_city][curr_name] = {}
                        continue
                    if cell_value == "" and curr_cell == 1:
                        continue
                    try:
                        result_dict[current_city][curr_name]['Phone'].append(cell_Value)
                    except:
                        result_dict[current_city][curr_name]['Phone'] = [cell_value]

The above function will return python dictionary as below -

{ 'New York' : { 'Tom' : [92929292, 33929] }, ........}

I will then traverse the directory and write new excel.

However, I want some generic way of splitting merged cells.

Upvotes: 3

Views: 12303

Answers (3)

Dristi
Dristi

Reputation: 1

import xlrd
import xlsxwriter
import numpy as np
import pandas as pd
def rep(l,i):
    j= i
    while(j>=0):
        if not l[j-1] == u'':
            return l[j-1]
        else:
            j = j-1
def write_df2xlsx(df,filename):
    # Create a Pandas Excel writer using XlsxWriter as the engine.
    writer = pd.ExcelWriter(filename,engine='xlsxwriter')

    # Convert the dataframe to an XlsxWriter Excel object.
    df.to_excel(writer, sheet_name='Sheet1', index = False)

    # Close the Pandas Excel writer and output the Excel file.
    writer.save()

def csv_from_excel(filename):

    wb = xlrd.open_workbook(filename)
    worksheet_names = wb.sheet_names()
    for worksheet_name in worksheet_names:
        sh = wb.sheet_by_name(worksheet_name)
        #To find the headers/column names of the xlsx file

        header_index = 0
        for i in range(sh.nrows):
            if(len(filter(lambda x: not (x.value == xlrd.empty_cell.value), sh.row(i))) == len(sh.row(i))):
                header_row = sh.row(i)
                header_index = i
                break
        columns = []
        for cell in range(len(header_row)):
            value = sh.cell_value(header_index, cell)
            columns.append(value)
        rows = []
        for rownum in range(header_index+1,sh.nrows):
            rows.append(sh.row_values(rownum))
        data = pd.DataFrame(rows,columns = columns)
        cols = [col for col in data.columns if u'' in list(data[col])]
        res = []
        for col in cols:
            t_list = list(data[col])
            res.append(map(lambda x,y: rep(list(data[col]),y[0]) if x == u'' else x,t_list,enumerate(t_list)))
        for (col,r) in zip(cols,res):
            data[col] = pd.core.series.Series(r)
        write_df2xlsx(data,'ResultFile.xlsx')   

Upvotes: 0

Emilio M Bumachar
Emilio M Bumachar

Reputation: 2613

This function gets the "real" cell value, i.e., the value of the merged cell if the coordinates are anywhere inside the merged cell.

def unmergedValue(rowx,colx,thesheet):
    for crange in thesheet.merged_cells:
        rlo, rhi, clo, chi = crange
        if rowx in xrange(rlo, rhi):
            if colx in xrange(clo, chi):
                return thesheet.cell_value(rlo,clo)
    #if you reached this point, it's not in any merged cells
    return thesheet.cell_value(rowx,colx)

Loosely based on http://www.lexicon.net/sjmachin/xlrd.html#xlrd.Sheet.merged_cells-attribute

Very innefficient, but should be acceptable for small-ish spreadsheets.

Upvotes: 1

thinker3
thinker3

Reputation: 13301

if your file has no empty cells in the middle, this may help, read the file, do some job, rewrite it.

def read_merged_xls(file_contents):
    book = xlrd.open_workbook(file_contents=file_contents)
    data = []
    sheet = book.sheet_by_index(0)
    for rx in range(sheet.nrows):
        line = [] 
        for ry in range(sheet.ncols):
            cell = sheet.cell_value(rx,ry)
            if not cell:
                cell = data[-1][ry] if data else ''
            line.append(cell)
        data.append(line)
    return data

Upvotes: 0

Related Questions