使用Pandas Styler格式化LaTeX中DataFrame表头遇属性错误求助

错误原因 #

你遇到的AttributeError: 'Series' object has no attribute 'columns'是因为apply_index要求传入的函数返回与目标索引长度匹配的样式序列，而你直接返回单个样式字符串，导致内部处理时出现类型不匹配的问题。

两种正确实现方式 #

方式一：修正apply_index的用法 #

调整函数，返回与列索引长度一致的样式列表，确保每个表头单元格都能应用样式：

import pandas as pd
import numpy as np

df = pd.DataFrame(np.random.rand(10,2)*5,
                  index=pd.date_range(start="2021-01-01", periods=10),
                  columns=["Tokyo", "Beijing"])

df.index.names = ['date']
df.reset_index(inplace=True)

def header_custom(v):
    # 返回与列索引长度匹配的样式列表
    return ["background-color: blue; color: black; font-weight: bold;"] * len(v)

# 针对列索引（表头）应用样式
styler = df.style.apply_index(header_custom, axis="columns")

with open('temp.tex','w') as file: 
    file.write(styler.to_latex(hrules=True, convert_css=True, 
                               column_format="p{30mm}|m|m|M|", 
                               multirow_align="c", multicol_align="c", 
                               clines='all;data'))

方式二：使用set_table_styles（推荐） #

通过CSS选择器直接定位表头单元格，语法更清晰易维护：

import pandas as pd
import numpy as np

df = pd.DataFrame(np.random.rand(10,2)*5,
                  index=pd.date_range(start="2021-01-01", periods=10),
                  columns=["Tokyo", "Beijing"])

df.index.names = ['date']
df.reset_index(inplace=True)

# 定义表头样式
styler = df.style.set_table_styles([
    {
        'selector': 'th.col_heading',  # 匹配所有列标题单元格
        'props': [
            ('background-color', 'blue'),
            ('color', 'black'),
            ('font-weight', 'bold')
        ]
    }
])

with open('temp.tex','w') as file: 
    file.write(styler.to_latex(hrules=True, convert_css=True, 
                               column_format="p{30mm}|m|m|M|", 
                               multirow_align="c", multicol_align="c", 
                               clines='all;data'))

注意事项 #

确保使用的pandas版本在1.3.0及以上，Styler.to_latex的convert_css、clines等参数是较新版本才支持的特性。

内容的提问来源于stack exchange，提问作者User981636