Pandas `to_latex`：如何格式化多索引表？

Question

问题

我正在尝试格式化 Pandas 表的to_latex输出，这是pivot_table操作的结果。 我的问题是：在这种情况下，如何将格式化程序传递给to_latex方法？

例子

更具体地说，考虑这个例子。

df = pd.DataFrame({"A": ["foo", "foo", "foo", "foo", "foo","bar", "bar", "bar", "bar"],
                   "B": ["one", "one", "one", "two", "two","one", "one", "two", "two"],
                   "C": ["small", "large", "large", "small","small", "large", "small","small","large"],
                   "D": [1, 2, 2, 3, 3, 4, 5, 6, 7],
                   "E": [2, 4, 5, 5, 6, 6, 8, 9, 9]})

table = pd.pivot_table(df, values=['D','E'], index=['A'], columns=['C','B'], aggfunc=np.sum)

输出是

    D                E
C   large   small   large   small
B   one     two     one     two     one     two     one     two
A                               
bar     4.0     7.0     5.0     6.0     6.0     9.0     8.0     9.0
foo     4.0     NaN     1.0     6.0     9.0     NaN     2.0     11.0 

我的尝试

在这里，我尝试以不同small large 。

    def f2(x):
        return f"%.{2}f" % x
    def f5(x):
        return f"%.{5}f" % x
    
    formatters = {'large':f2,'small':f5}
    table.to_latex('tables/tab.tex',formatters=formatters,longtable=True)

但是结果表的格式不正确。 我的猜测是它与这张桌子的分层性质有关。 我能做些什么？

Answer 1

尝试这个。 这应该有效。

df = pd.DataFrame({"A": ["foo", "foo", "foo", "foo", "foo","bar", "bar", "bar", "bar"],
                   "B": ["one", "one", "one", "two", "two","one", "one", "two", "two"],
                   "C": ["small", "large", "large", "small","small", "large", "small","small","large"],
                   "D": [1, 2, 2, 3, 3, 4, 5, 6, 7],
                   "E": [2, 4, 5, 5, 6, 6, 8, 9, 9]})

table = pd.pivot_table(df, values=['D','E'], index=['A'], columns=['C','B'], aggfunc=np.sum).reset_index()

def f2(x):
        return f"%.{2}f" % x

def f5(x):
    return f"%.{5}f" % x

formatters = {'large':f2,'small':f5}

table.to_latex('tables/tab.tex',formatters=formatters,longtable=True)