Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Pandas DataFrame to Excel: Vertical Alignment of Index

Given the following data frame: import pandas as pd

d=pd.DataFrame({'a':['a','a','b','b'],
               'b':['a','b','c','d'],
               'c':[1,2,3,4]})
d=d.groupby(['a','b']).sum()
d

enter image description here

I'd like to export this with the same alignment with respect to the index (see how the left-most column is centered vertically?). The rub is that when exporting this to Excel, the left column is aligned to the top of each cell:

writer = pd.ExcelWriter('pandas_out.xlsx', engine='xlsxwriter')
workbook  = writer.book
f=workbook.add_format({'align': 'vcenter'})
d.to_excel(writer, sheet_name='Sheet1')
writer.save()

...produces...

enter image description here

Is there any way to center column A vertically via XLSX Writer or another library?

Thanks in advance!

like image 932
Dance Party2 Avatar asked Dec 28 '16 14:12

Dance Party2


Video Answer


1 Answers

You are trying to change the formatting of the header so you should first reset the default header settings

from pandas.io.formats.excel import ExcelFormatter
ExcelFormatter.header_style = None

Then apply the formatting as required

format = workbook.add_format()
format.set_align('center')
format.set_align('vcenter')

worksheet.set_column('A:C',5, format)

here is complete working code

d=pd.DataFrame({'a':['a','a','b','b'],
               'b':['a','b','c','d'],
               'c':[1,2,3,4]})
d=d.groupby(['a','b']).sum()

pd.core.format.header_style = None

writer = pd.ExcelWriter('pandas_out.xlsx', engine='xlsxwriter')
workbook  = writer.book
d.to_excel(writer, sheet_name='Sheet1')

worksheet = writer.sheets['Sheet1']

format = workbook.add_format()
format.set_align('center')
format.set_align('vcenter')

worksheet.set_column('A:C',5, format)
writer.save()
like image 147
Shijo Avatar answered Oct 17 '22 03:10

Shijo