我们如何为 pandas 数据框的 headers 列和行着色？

Question

如何在我想以 excel 格式保存的数据框 table 中显示不同颜色的列标题，我之前将其分组为 5？

Answer 1

假设您有以下格式的数据框。

import pandas as pd
import numpy as np

data = pd.DataFrame(np.random.rand(5, 5), columns=[
                    'col_' + str(e) for e in range(5)], index=['col_' + str(e) for e in range(5)])
data

          col_0     col_1     col_2     col_3     col_4
col_0  0.196830  0.306508  0.515714  0.033282  0.640673
col_1  0.278842  0.189177  0.616362  0.577048  0.805790
col_2  0.699674  0.251704  0.146142  0.144638  0.882772
col_3  0.794672  0.748220  0.780501  0.716122  0.278373
col_4  0.535306  0.182256  0.662058  0.323910  0.908328

我们先导出它。

data.to_excel('data.xlsx')

现在，如果您已经知道要更改哪些列名称，则可以执行以下操作：

group_1 = ['col_0', 'col_3', 'col_4']
group_2 = ['col_1', 'col_2']

def change_color(workbook_param, color_hex_code):
    """Returns color format for excelsheet."""
    header_format = workbook_param.add_format({
        'bold': True,
        'text_wrap': True,
        'valign': 'top',
        'fg_color': color_hex_code,
        'border': 1})

    return header_format


group_1_ind = [data.columns.to_list().index(patient) for patient in group_1]
group_1_ind  # [0, 3, 4]
group_2_ind = [data.columns.to_list().index(patient) for patient in group_2]
group_2_ind  # [1, 2]

group_1_data = data[group_1].copy()
group_2_data = data[group_2].copy()

# Create a Pandas Excel writer.
writer = pd.ExcelWriter('data.xlsx', engine='xlsxwriter')

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

# Get the xlsxwriter workbook and worksheet objects. You can change sheet name as well.
workbook = writer.book
worksheet = writer.sheets['Sheet1']

# Do the modifications for desired columns.
for col_num, value in zip(group_1_ind, group_1_data.columns.values):
    worksheet.write(0, col_num + 1, value, change_color(workbook, '#e3fc03'))

for col_num, value in zip(group_2_ind, group_2_data.columns.values):
    worksheet.write(0, col_num + 1, value, change_color(workbook, '#eb4034'))
    
    
# Do the modifications for desired indexes.
for ind_num, value in zip(group_1_ind, group_1_data.columns.values):
    worksheet.write(ind_num + 1, 0, value, change_color(workbook, '#e3fc03'))

for ind_num, value in zip(group_2_ind, group_2_data.columns.values):
    worksheet.write(ind_num + 1, 0, value, change_color(workbook, '#eb4034'))


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

现在您的 excel 文件将如下所示：

您可以优化代码以便进一步修改以轻松处理。

我们如何为 pandas 数据框的 headers 列和行着色？

How can we color the column and row headers of pandas dataframe?

python

excel

pandas

background-color