Создание отформатированной таблицы из списка кортежей - PullRequest
0 голосов
/ 27 ноября 2018

У меня есть список данных, который выглядит примерно так:

[
    (
        1,
        u'python -c \'print("ok")\'',
        u'data',
        u'python'
    ), (
        2,
        u'python -c \'print("this is some data")\'',
        u'data',
        u'python'
    )
]

Эти данные извлекаются из базы данных и отображаются как это, и они постоянно растут.То, что я хотел бы сделать, это отобразить данные следующим образом:

Language  |  Type  |   Payload
-------------------------------
 python   |  data  |  python -c 'print("ok")'
 python   |  data  |  python -c 'print("this is some data")'

У меня есть функция, которая делает то же самое, но это не совсем так, как ожидалось:

def print_table(data, cols, width):
    n, r = divmod(len(data), cols)
    pattern = "{{:{}}}".format(width)
    line = "\n".join(pattern * cols for _ in range(n))
    last_line = pattern * r
    print(line.format(*data))
    print(last_line.format(*data[n*cols]))

Какя могу получить вывод моих данных, чтобы выглядеть как хотел?Из ответов это возможно с pandas, но я также хотел бы сделать это без установки внешних модулей

Ответы [ 3 ]

0 голосов
/ 27 ноября 2018

Анализируйте данные на предмет максимальной ширины и используйте форматирование строки - некоторое «творческое» форматирование позже:

data = [
    (
        1,
        u'python -c \'print("ok")\'',
        u'data',
        u'python'
    ), (
        2,
        u'python -c \'print("this is some data")\'',
        u'data',
        u'python'
    )
]



def print_table(data):
    widths = {0:0, 3:len("Language"),2:len("Type"),1:len("Payload")}
    for k in data:
        for i,d in enumerate(k): 
            widths[i] = max(widths[i],len(str(d))) 
    # print(widths)

    lan, typ, pay = ["Language","Type","Payload"]
    print(f"{lan:<{widths[3]}}  |  {typ:<{widths[2]}}  |  {pay:<{widths[1]}}")
    # adjust by 10 for '  |  ' twice
    print("-" * (widths[1]+widths[2]+widths[3]+10)) 
    for k in data:
        _, pay, typ, lan = k
        print(f"{lan:<{widths[3]}}  |  {typ:<{widths[2]}}  |  {pay:<{widths[1]}}") 

Вывод:

Language  |  Type  |  Payload                               
------------------------------------------------------------
python    |  data  |  python -c 'print("ok")'               
python    |  data  |  python -c 'print("this is some data")'

Эквивалентный код Python 2.7:

# w == widths - would break 79 chars/line else wise
def print_table(data):
    w = {0:0, 3:len("Language"),2:len("Type"),1:len("Payload")}
    for k in data:
        for i,d in enumerate(k): 
            w[i] = max(w[i],len(str(d))) 


    lan, typ, pay = ["Language","Type","Payload"]
    print "{:<{}}  |  {:<{}}  |  {:<{}}".format(lan, w[3], typ, w[2], pay, w[1])   
    print "-" * (w[1]+w[2]+w[3]+10) 
    for k in data:
        _, pay, typ, lan = k
        print "{:<{}}  |  {:<{}}  |  {:<{}}".format(lan, w[3], typ, w[2], pay, w[1]) 
0 голосов
/ 27 ноября 2018

Решение, которое обрабатывает любое количество столбцов:

from operator import itemgetter

data = [
    ('ID', 'Payload', 'Type', 'Language'),
    (1, u'python -c \'print("ok")\'', u'data', u'python'),
    (2, u'python -c \'print("this is some data")\'', u'data', u'python')
]


def print_table(data):
    lengths = [
        [len(str(x)) for x in row]
        for row in data
    ]

    max_lengths = [
        max(map(itemgetter(x), lengths))
        for x in range(0, len(data[0]))
    ]

    format_str = ''.join(map(lambda x: '%%-%ss | ' % x, max_lengths))

    print(format_str % data[0])
    print('-' * (sum(max_lengths) + len(max_lengths) * 3 - 1))

    for x in data[1:]:
        print(format_str % x)

print_table(data)

Вывод:

$ python table.py
ID | Payload                                | Type | Language |
---------------------------------------------------------------
1  | python -c 'print("ok")'                | data | python   |
2  | python -c 'print("this is some data")' | data | python   |
0 голосов
/ 27 ноября 2018

Вы можете использовать pandas для этого:

import pandas as pd
data = pd.DataFrame(a, columns=['id','Payload', 'type', 'Language'])
print(data)

дает вам:

   id                                 Payload  type Language
0   1                 python -c 'print("ok")'  data   python
1   2  python -c 'print("this is some data")'  data   python
...