Khá In một khung dữ liệu gấu trúc

Question

Làm cách nào để in một khung dữ liệu gấu trúc làm bảng dựa trên văn bản đẹp, như sau?

+------------+---------+-------------+ 
| column_one | col_two | column_3 | 
+------------+---------+-------------+ 
|   0 | 0.0001 | ABCD  | 
|   1 | 1e-005 | ABCD  | 
|   2 | 1e-006 | long string | 
|   3 | 1e-007 | ABCD  | 
+------------+---------+-------------+ 

Cập nhật: Tôi tìm thấy giải pháp cho điều này, được đăng dưới dạng câu trả lời bên dưới.

Answer 1

Bạn có thể sử dụng prettytable để hiển thị bảng dưới dạng văn bản. Bí quyết là chuyển đổi data_frame thành tệp csv trong bộ nhớ và có thể đọc được nó. Đây là mã:

from StringIO import StringIO 
import prettytable  

output = StringIO() 
data_frame.to_csv(output) 
output.seek(0) 
pt = prettytable.from_csv(output) 
print pt 

Answer 2

Tôi đã sử dụng câu trả lời của Ofer một thời gian và thấy nó tuyệt vời trong hầu hết các trường hợp. Thật không may, do sự không nhất quán giữa pandas's to_csv và prettytable 's from_csv, tôi đã phải sử dụng một cách khác theo cách khác.

Một trường hợp thất bại là một dataframe chứa dấu phẩy:

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

Prettytable đặt ra một lỗi có dạng:

Error: Could not determine delimiter 

Chức năng sau xử lý trường hợp này:

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

Nếu bạn không quan tâm đến chỉ mục, sử dụng:

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

Answer 3

Tôi vừa tìm thấy công cụ tuyệt vời cho nhu cầu đó, nó được gọi là tabulate.

Nó in dữ liệu dạng bảng và hoạt động với 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  | 
+----+-----------+-------------+ 

Lưu ý: Có một mở Pull request mà sẽ cho phép để bao gồm/loại trừ các chỉ số.