如何使用pandas为整个列包装文本?

时间:2017-08-31 15:39:17

标签: python excel python-3.x pandas

我想使用pandas来包装整个列。我已经设置了列的宽度,我只需要将整个列包装起来,因为它们都在一行中。

我搜索不同的答案并不直截了当。我不需要编辑列的宽度,我只想将文本包装在单元格中,这与我突出显示列n点击" Wrap Text"。但我想使用Python脚本来实现它,因此我想使用pandas来实现它。

所以基本上是

before looks like this

after looks like this

但是我想通过pandas这样做,而不是通过点击换行文本手动完成。我想将包裹应用于整个列而不仅仅是一个单元格。

1 个答案:

答案 0 :(得分:4)

您可以使用解决方案修改example_pandas_column_formats

import string

long_text = 'aa aa ss df fff ggh ttr tre ww rr tt ww errr t ttyyy eewww rr55t e'
data = {'a':[long_text, long_text, 'a'],'c':[long_text,long_text,long_text],
        'b':[1,2,3]}       
df = pd.DataFrame(data)
#choose columns of df for wrapping
cols_for_wrap = ['a','c']


writer = pd.ExcelWriter('aaa.xlsx', engine='xlsxwriter')
df.to_excel(writer, sheet_name='Sheet1', index=False)

#modifyng output by style - wrap
workbook  = writer.book
worksheet = writer.sheets['Sheet1']
wrap_format = workbook.add_format({'text_wrap': True})

#dictionary for map position of selected columns to excel headers
d = dict(zip(range(26), list(string.ascii_uppercase)))
print (d)
{0: 'A', 1: 'B', 2: 'C', 3: 'D', 4: 'E', 5: 'F', 6: 'G', 7: 'H', 8: 'I', 
9: 'J', 10: 'K', 11: 'L', 12: 'M', 13: 'N', 14: 'O', 15: 'P', 16: 'Q', 
17: 'R', 18: 'S', 19: 'T', 20: 'U', 21: 'V', 22: 'W', 23: 'X', 24: 'Y', 25: 'Z'}

#get positions of columns
for col in df.columns.get_indexer(cols_for_wrap):
#map by dict to format like "A:A"     
    excel_header  =  d[col] + ':' + d[col]
    #None means not set with
    worksheet.set_column(excel_header, None, wrap_format)
    #for with = 20
    #worksheet.set_column(excel_header, 20, wrap_format)

writer.save()