如何使用Styler.to_latex将列名与多级索引名输出在同一行？

问题描述 #

使用df.style.to_latex()导出带多级索引的DataFrame时，默认会将索引名和列名分两行输出：

&  & A & B & C & D \\
alpha & beta &  &  &  &  \\

需求是将索引名与列名合并为同一行：

alpha & beta & A & B & C & D \\

且不想通过重置索引（会重复索引值，无法使用\multirow）的方式实现。

解决方案 #

方法1：直接修改LaTeX输出内容 #

利用style.to_latex()生成内容后，替换默认的两行表头为自定义单行表头：

import numpy as np
import pandas as pd

arrays = [
    np.array(["bar", "bar", "baz", "baz", "foo", "foo", "qux", "qux"]),
    np.array(["one", "two", "one", "two", "one", "two", "one", "two"]),
]
df = pd.DataFrame(np.random.randn(8, 4), index=arrays, columns=["A", "B", "C", "D"])
df.index.names = ("alpha", "beta")

# 生成默认LaTeX内容
default_latex = df.style.to_latex()

# 构造目标表头行
target_header = f"{df.index.names[0]} & {df.index.names[1]} & {' & '.join(df.columns)} \\\\"

# 替换原有的两行表头
final_latex = default_latex.replace(
    r" &  & A & B & C & D \\" + "\n" + r"alpha & beta &  &  &  &  \\",
    target_header
)

print(final_latex)

方法2：使用df.to_latex()自定义表头 #

直接使用df.to_latex()生成无表头的表格内容，再手动添加自定义表头：

import numpy as np
import pandas as pd

arrays = [
    np.array(["bar", "bar", "baz", "baz", "foo", "foo", "qux", "qux"]),
    np.array(["one", "two", "one", "two", "one", "two", "one", "two"]),
]
df = pd.DataFrame(np.random.randn(8, 4), index=arrays, columns=["A", "B", "C", "D"])
df.index.names = ("alpha", "beta")

# 生成不带表头的表格内容，保留多级索引的multirow格式
table_content = df.to_latex(index=True, header=False, multirow=True)

# 构造自定义表头
header_row = f"{df.index.names[0]} & {df.index.names[1]} & {' & '.join(df.columns)} \\\\"

# 将表头插入到表格开头
final_latex = table_content.replace(
    r"\begin{tabular}{llrrrr}",
    r"\begin{tabular}{llrrrr}" + "\n" + header_row
)

print(final_latex)

两种方法都能实现目标效果，且保留多级索引的\multirow格式，不会重复索引值。

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