Печатать (табулировать (...)), чтобы красиво печатать многоиндексных панд?

Скажем, у меня есть фрейм данных панд:

import numpy as np
import pandas as pd
from tabulate import tabulate

A = pd.DataFrame(np.random.randint(0,10,(3,6)), index= ['uno', 'dos', 'tres'])
A.columns = ['A','B','C','D','E','F']
A.index.names = ['type']
A.columns.names= ['group']

h = [A.index.names[0] +'/'+ A.columns.names[0]] + list(A.columns)
print(tabulate(A, headers= h, tablefmt= 'grid'))

который дает:

+--------------+-----+-----+-----+-----+-----+-----+
| type/group   |   A |   B |   C |   D |   E |   F |
+==============+=====+=====+=====+=====+=====+=====+
| uno          |   3 |   1 |   6 |   0 |   7 |   0 |
+--------------+-----+-----+-----+-----+-----+-----+
| dos          |   9 |   5 |   3 |   0 |   6 |   6 |
+--------------+-----+-----+-----+-----+-----+-----+
| tres         |   6 |   7 |   4 |   6 |   8 |   4 |
+--------------+-----+-----+-----+-----+-----+-----+

Теперь добавляем слой:

iterable = [['A', 'B'], ['AA', 'BB', 'CC']]
A.columns = pd.MultiIndex.from_product(iterable,
                                       names= ['group', 'subgroup'])
A.index.names = ['type']

даст с помощью оператора print:

group     A        B      
subgroup AA BB CC AA BB CC
type                      
uno       3  1  6  0  7  0
dos       9  5  3  0  6  6
tres      6  7  4  6  8  4

Как правило, это плохо отражается на документе, таком как pweave.

Как я могу использовать print(tabulate(...)), например, чтобы каждая группа и подгруппа отображались в отдельной строке?

Спасибо

1 ответ

Решение

С табличной версией 0.8.1 или новее,

import numpy as np
import pandas as pd
from tabulate import tabulate

A = pd.DataFrame(np.random.randint(0,10,(3,6)), index= ['uno', 'dos', 'tres'])
A.columns = ['A','B','C','D','E','F']
A.index.names = ['type']
A.columns.names= ['group']

iterable = [['A', 'B'], ['AA', 'BB', 'CC']]
A.columns = pd.MultiIndex.from_product(iterable, names= ['group', 'subgroup'])

h = [A.index.names[0] +'/'+ A.columns.names[0]] + list(map('\n'.join, A.columns.tolist()))
print(tabulate(A, headers= h, tablefmt= 'grid'))

доходность

+--------------+------+------+------+------+------+------+
| type/group   |    A |    A |    A |    B |    B |    B |
|              |   AA |   BB |   CC |   AA |   BB |   CC |
+==============+======+======+======+======+======+======+
| uno          |    8 |    4 |    3 |    3 |    4 |    2 |
+--------------+------+------+------+------+------+------+
| dos          |    1 |    1 |    1 |    9 |    8 |    9 |
+--------------+------+------+------+------+------+------+
| tres         |    0 |    4 |    3 |    6 |    1 |    7 |
+--------------+------+------+------+------+------+------+
Другие вопросы по тегам