Довольно печать фрейма данных pandas

115

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

+------------+---------+-------------+
| column_one | col_two |   column_3  |
+------------+---------+-------------+
|          0 |  0.0001 | ABCD        |
|          1 |  1e-005 | ABCD        |
|          2 |  1e-006 | long string |
|          3 |  1e-007 | ABCD        |
+------------+---------+-------------+
Офер
источник

Ответы:

181

Я только что нашел для этого отличный инструмент, он называется табуляция .

Он печатает табличные данные и работает с DataFrame.

from tabulate import tabulate
import pandas as pd

df = pd.DataFrame({'col_two' : [0.0001, 1e-005 , 1e-006, 1e-007],
                   'column_3' : ['ABCD', 'ABCD', 'long string', 'ABCD']})
print(tabulate(df, headers='keys', tablefmt='psql'))

+----+-----------+-------------+
|    |   col_two | column_3    |
|----+-----------+-------------|
|  0 |    0.0001 | ABCD        |
|  1 |    1e-05  | ABCD        |
|  2 |    1e-06  | long string |
|  3 |    1e-07  | ABCD        |
+----+-----------+-------------+

Примечание:

Чтобы подавить индексы строк для всех типов данных, передайте showindex="never"или showindex=False.

Ромен
источник
5
Если у вас нет доступа к переднему краю, вы можете вообще tabulate([list(row) for row in df.values], headers=list(df.columns))избавиться от индекса
Педро Дуарте
1
Не очень хорошо работает, когда у вас есть иерархии в индексах строк и столбцах.
Сиддхарт
Убедитесь, что вы это делаете, print(tabulate(df, **kwargs))а не просто tabulate(df, **kwargs); последний покажет все новые строчки \n....
Dror
6
Чтобы подавить левый столбец индекса, можно также добавитьshowindex=False
Артур
19

панды> = 1.0

Если вы хотите, чтобы встроенная функция сбрасывала ваши данные в какую-то уценку github, теперь она у вас есть. Взгляните на to_markdown:

df = pd.DataFrame({"A": [1, 2, 3], "B": [1, 2, 3]}, index=['a', 'a', 'b'])  
print(df.to_markdown()) 

|    |   A |   B |
|:---|----:|----:|
| a  |   1 |   1 |
| a  |   2 |   2 |
| b  |   3 |   3 |

Вот как это выглядит на github:

введите описание изображения здесь

Обратите внимание, что вам все равно потребуется установить tabulateпакет.

cs95
источник
17

Если вы работаете в записной книжке Jupyter, вы можете запустить следующий код для интерактивного отображения фрейма данных в хорошо отформатированной таблице.

Этот ответ основан на ответе to_html ('temp.html') выше, но вместо создания файла отображает хорошо отформатированную таблицу непосредственно в записной книжке:

from IPython.display import display, HTML

display(HTML(df.to_html()))

Кредит для этого кода из-за примера в: Показать DataFrame как таблицу в iPython Notebook

Марк Андерсен
источник
15

Вы можете использовать prettytable для отображения таблицы как текста. Хитрость заключается в том, чтобы преобразовать data_frame в файл csv в памяти и получить удобное чтение. Вот код:

from StringIO import StringIO
import prettytable    

output = StringIO()
data_frame.to_csv(output)
output.seek(0)
pt = prettytable.from_csv(output)
print pt
Офер
источник
Какая это была версия панд?
WAF
4
AFAIK, prettytableв основном считается заброшенным ПО. Жалко тоже, потому что это была хорошая посылка. :(
dmn
@dmn, значит, он больше не поддерживается?
мюон
prettytableне имеет релиза с 6 апреля 2013 года. tabulateявляется его духовным предшественником и имеет регулярные релизы, последний из которых - 24 января 2019 года.
noddy
7

Некоторое время я использовал ответ Офера и в большинстве случаев находил его отличным. К сожалению, из-за несоответствий между to_csv pandas и from_csv prettytable мне пришлось использовать prettytable по-другому.

Один случай сбоя - это кадр данных, содержащий запятые:

pd.DataFrame({'A': [1, 2], 'B': ['a,', 'b']})

Prettytable вызывает ошибку формы:

Error: Could not determine delimiter

Следующая функция обрабатывает этот случай:

def format_for_print(df):    
    table = PrettyTable([''] + list(df.columns))
    for row in df.itertuples():
        table.add_row(row)
    return str(table)

Если вас не волнует индекс, используйте:

def format_for_print2(df):    
    table = PrettyTable(list(df.columns))
    for row in df.itertuples():
        table.add_row(row[1:])
    return str(table)
EJRB
источник
Привет, format_for_print()функция, похоже, не печатает индекс Pandas DataFrame. Я установил индекс с помощью, df.index.name = 'index'но это не печатает столбец индекса с именем.
edesz
2

Следуя ответу Марка, если вы по какой-то причине не используете Jupyter, например, вы хотите провести быстрое тестирование на консоли, вы можете использовать DataFrame.to_stringметод, который работает, по крайней мере, с Pandas 0.12 (2014) и далее. .

import pandas as pd

matrix = [(1, 23, 45), (789, 1, 23), (45, 678, 90)]
df = pd.DataFrame(matrix, columns=list('abc'))
print(df.to_string())

#  outputs:
#       a    b   c
#  0    1   23  45
#  1  789    1  23
#  2   45  678  90
сигинта
источник
0

Может быть, вы ищете что-то вроде этого:

def tableize(df):
    if not isinstance(df, pd.DataFrame):
        return
    df_columns = df.columns.tolist() 
    max_len_in_lst = lambda lst: len(sorted(lst, reverse=True, key=len)[0])
    align_center = lambda st, sz: "{0}{1}{0}".format(" "*(1+(sz-len(st))//2), st)[:sz] if len(st) < sz else st
    align_right = lambda st, sz: "{0}{1} ".format(" "*(sz-len(st)-1), st) if len(st) < sz else st
    max_col_len = max_len_in_lst(df_columns)
    max_val_len_for_col = dict([(col, max_len_in_lst(df.iloc[:,idx].astype('str'))) for idx, col in enumerate(df_columns)])
    col_sizes = dict([(col, 2 + max(max_val_len_for_col.get(col, 0), max_col_len)) for col in df_columns])
    build_hline = lambda row: '+'.join(['-' * col_sizes[col] for col in row]).join(['+', '+'])
    build_data = lambda row, align: "|".join([align(str(val), col_sizes[df_columns[idx]]) for idx, val in enumerate(row)]).join(['|', '|'])
    hline = build_hline(df_columns)
    out = [hline, build_data(df_columns, align_center), hline]
    for _, row in df.iterrows():
        out.append(build_data(row.tolist(), align_right))
    out.append(hline)
    return "\n".join(out)


df = pd.DataFrame([[1, 2, 3], [11111, 22, 333]], columns=['a', 'b', 'c'])
print tableize(df)
Вывод:
+ ------- + ---- + ----- +
| а | б | c |
+ ------- + ---- + ----- +
| 1 | 2 | 3 |
| 11111 | 22 | 333 |
+ ------- + ---- + ----- +
Пафконе
источник
-6

Мне нужна была бумажная распечатка фрейма данных, но я хотел добавить некоторые результаты и комментарии на той же странице. Я проработал все вышесказанное и не смог получить то, что хотел. В итоге я использовал операторы file.write (df1.to_csv ()) и file.write (",,, blah ,,,,,, blah"), чтобы добавить на страницу свои дополнения. Когда я открыл файл csv, он сразу попал в электронную таблицу, которая распечатала все в нужном темпе и формате.

Джон
источник