[파이썬/python] tabulate, 테이블 포맷 출력

일을 하다 보면 테이블을 이용해서 보고하는 경우가 많다.

데이터 통계를 낼 수도 있고, 성능을 정리해서 올릴 때도 있고.

 

주로 사내 깃헙(oss)에 올려서 공유를 하는데,

깔끔하게 정리하지 않아서 테이블이 꼬이면 마크다운이라 어디가 틀린지 찾기 어렵다...

 

그래서 주로 tabulate를 사용하고 있는데 항상 헷갈려서 간단히 예제로 정리해 본다.

설치는 간단히 pip로 설치한다.

 

• 설치

$ pip install tabulate

 

• 예제

from tabulate import tabulate

headers=['model', 'EM', 'F1 score']
output = [
    ['BERT', 84.04, 92.43],
    ['ROBERT', 87.62, 94.61]
]

perf_table = tabulate(output, headers=headers, tablefmt="github")
print(perf_table)

tablefmt="github"으로 주어야 바로 markdown에 적용이 된다.

 

• 출력 결과

| model   |    EM |   F1 score |
|---------|-------|------------|
| BERT    | 84.04 |      92.43 |
| ROBERT  | 87.62 |      94.61 |

복사 붙여넣기로 성능표가 완성된다. 성능표은 korquad dev셋에서의 성능이다.

 

외에도 다양한 포맷들이 있으니 상황에 따라서 알맞게 사용하자.

• "plain"
• "simple"
• "github"
• "grid"
• "fancy_grid"
• "pipe"
• "orgtbl"
• "jira"
• "presto"
• "pretty"
• "psql"
• "rst"
• "mediawiki"
• "moinmoin"
• "youtrack"
• "html"
• "unsafehtml"
• "latex"
• "latex_raw"
• "latex_booktabs"
• "latex_longtable"
• "textile"
• "tsv"

 

+ 간단한 markdown 테이블을 만들어주는 웹사이트도 있다.