如何为Pandas中转置后的DataFrame指定LaTeX输出的行格式？

问题背景 #

对DataFrame转置（df.T）后，无法通过常规的列指定格式方式设置行的数字格式，且转置后原本的整数列会被转为浮点数，导致LaTeX输出不符合预期。

解决方案：利用subset参数指定行格式 #

Pandas的style.format支持通过subset参数精准定位需要设置格式的行（或列），你可以通过行索引+列范围的组合，为转置后的特定行设置对应格式。

完整代码示例

import pandas as pd

# 构建原始DataFrame
values = [{"id": "id1", "c1": 1e-10, "c2": int(1000)}]
df = pd.DataFrame.from_dict(values).set_index("id")
# 转置DataFrame
dft = df.T

# 针对转置后的行设置格式
style = dft.style
# 为c1行的所有列设置两位小数格式
style.format("{:.2f}", subset=("c1", :))
# 为c2行的所有列设置整数格式
style.format("{:d}", subset=("c2", :))

# 输出LaTeX代码
print(style.to_latex())

预期输出

\begin{tabular}{lr}
id & id1 \\
c1 & 0.00 \\
c2 & 1000 \\
\end{tabular}

原理说明 #

1. subset参数的用法：subset=(行索引, 列索引)，其中:代表选择所有列（或行）。这里通过("c1", :)选中c1行的所有列，再应用对应的格式字符串。
2. 整数格式强制修正：转置后Pandas会将混合类型的列统一为浮点数，通过"{:d}"格式字符串可以强制将浮点数格式化为整数显示，避免出现1000.000000的无效格式。

另一种思路：先处理数据类型再格式化 #

如果需要先确保转置后的行数据类型正确，也可以先转换数据类型再格式化：

dft = df.T
# 将c2行转换为整数类型
dft.loc["c2"] = dft.loc["c2"].astype(int)
# 根据行索引判断并应用对应格式
style = dft.style.format(lambda x: "{:.2f}".format(x) if x.name == "c1" else "{:d}".format(x))
print(style.to_latex())

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