KevinMidboe/tableprint
1
0
Fork 0
mirror of https://github.com/KevinMidboe/tableprint.git synced 2025-10-29 18:00:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

adding csv function

Browse Source
This commit is contained in:
Niru Maheswaranathan
2017-06-08 16:58:27 -07:00
parent ac2accb231
commit 64884f1830
1 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tableprint/printer.py
View File

@@ -13,6 +13,7 @@ Usage
from __future__ import print_function, unicode_literals from __future__ import print_function, unicode_literals
import sys import sys
import csv
from numbers import Number from numbers import Number
import numpy as np import numpy as np
@@ -21,7 +22,7 @@ from six import string_types
from .style import LineStyle, STYLES from .style import LineStyle, STYLES
from .utils import ansi_len, format_line, parse_width from .utils import ansi_len, format_line, parse_width
__all__ = ('table', 'header', 'row', 'hrule', 'top', 'bottom', 'banner', 'dataframe', 'TableContext') __all__ = ('table', 'header', 'row', 'hrule', 'top', 'bottom', 'banner', 'dataframe', 'TableContext', 'csvfile')
STYLE = 'round' STYLE = 'round'
WIDTH = 11 WIDTH = 11
@@ -269,3 +270,10 @@ def dataframe(df, **kwargs):
A pandas DataFrame with the table to print A pandas DataFrame with the table to print
""" """
table(np.array(df), list(df.columns), **kwargs) table(np.array(df), list(df.columns), **kwargs)
def csvfile(filename, width=WIDTH, style=STYLE, format_spec=FMT):
"""Prints data from the given csvfile as a table"""
with open(filename, 'r') as file_handle:
headers, *data = list(csv.reader(file_handle))
table([list(map(float, row)) for row in data], headers=headers, width=width, style=style, format_spec=format_spec)