使用tabulate打印多索引数据帧

Question

如何打印如下所示的多索引数据帧：

import numpy as np
import tabulate
import pandas as pd


df = pd.DataFrame(np.random.randn(4, 3),
                  index=pd.MultiIndex.from_product([["foo", "bar"],
                                                    ["one", "two"]]),
                  columns=list("ABC"))

因此，Multindex的两个级别显示为单独的列，几乎与pandas本身打印它的方式相同：

In [16]: df
Out[16]: 
                A         B         C
foo one -0.040337  0.653915 -0.359834
    two  0.271542  1.328517  1.704389
bar one -1.246009  0.087229  0.039282
    two -1.217514  0.721025 -0.017185

但是，打印列表如下所示：

In [28]: print(tabulate.tabulate(df, tablefmt="github", headers="keys", showindex="always"))
|                |          A |         B |          C |
|----------------|------------|-----------|------------|
| ('foo', 'one') | -0.0403371 | 0.653915  | -0.359834  |
| ('foo', 'two') |  0.271542  | 1.32852   |  1.70439   |
| ('bar', 'one') | -1.24601   | 0.0872285 |  0.039282  |
| ('bar', 'two') | -1.21751   | 0.721025  | -0.0171852 |

Answer 1

MultiIndexes在内部是由元组表示的，所以tabulate向您展示了正确的东西。

如果你想要像列一样的显示，最简单的方法是先reset_index：

print(tabulate.tabulate(df.reset_index().rename(columns={'level_0':'', 'level_1': ''}), tablefmt="github", headers="keys", showindex=False))

输出：

|     |     |         A |         B |         C |
|-----|-----|-----------|-----------|-----------|
| foo | one | -0.108977 |  2.03593  |  1.11258  |
| foo | two |  0.65117  | -1.48314  |  0.391379 |
| bar | one | -0.660148 |  1.34875  | -1.10848  |
| bar | two |  0.561418 |  0.762137 |  0.723432 |

或者，您可以将MultiIndex修改为单个索引：

df2 = df.copy()
df2.index = df.index.map(lambda x: '|'.join(f'{e:>5} ' for e in x))

print(tabulate.tabulate(df2.rename_axis('index'), tablefmt="github", headers="keys", showindex="always"))

输出：

| index      |         A |         B |         C |
|------------|-----------|-----------|-----------|
| foo |  one | -0.108977 |  2.03593  |  1.11258  |
| foo |  two |  0.65117  | -1.48314  |  0.391379 |
| bar |  one | -0.660148 |  1.34875  | -1.10848  |
| bar |  two |  0.561418 |  0.762137 |  0.723432 |