python prettytable 打印表格

日常工作中会使用脚本统计数据,但数据输出的格式展示经常让人头疼,要自己写函数来实现。

于是为了省时间和避免重复“造车轮”, 找到了 prettytable 这个模块,github 地址

一、安装


安装很简单,执行以下命令:

pip install prettytable

readme 文档:https://github.com/jazzband/prettytable/blob/master/README

二、示例


import prettytable as pt

# tb = pt.PrettyTable( ["City name", "Area", "Population", "Annual Rainfall"])
tb = pt.PrettyTable()
tb.field_names = ["City name", "Area", "Population", "Annual Rainfall"]
tb.add_row(["Adelaide",1295, 1158259, 600.5])
tb.add_row(["Brisbane",5905, 1857594, 1146.4])
tb.add_row(["Darwin", 112, 120900, 1714.7])
tb.add_row(["Hobart", 1357, 205556,619.5])

print(tb)

输出:

+-----------+------+------------+-----------------+
| City name | Area | Population | Annual Rainfall |
+-----------+------+------------+-----------------+
|  Adelaide | 1295 |  1158259   |      600.5      |
|  Brisbane | 5905 |  1857594   |      1146.4     |
|   Darwin  | 112  |   120900   |      1714.7     |
|   Hobart  | 1357 |   205556   |      619.5      |
+-----------+------+------------+-----------------+

如果没有添加表头,那么会以默认的Field+编号显示,例如:

+---------+----------+----------+------------+
| Field 1 | Field 2  | Field 3  |  Field 4   |
+---------+----------+----------+------------+

三、添加数据


1、添加行 table.add_row()

在上面简单的示例中,我们就是按行添加数据的。
添加的数据必须要是列表(list)的形式,而且数据的列表长度要和表头的长度一样。

tb.add_row(["Hobart", 1357, 205556,619.5])

2、添加列 table.add_column()

添加一列也很简单,直接在原代码中添加:

import prettytable as pt

# tb = pt.PrettyTable( ["City name", "Area", "Population", "Annual Rainfall"])
tb = pt.PrettyTable()
tb.field_names = ["City name", "Area", "Population", "Annual Rainfall"]
tb.add_row(["Adelaide",1295, 1158259, 600.5])
tb.add_row(["Brisbane",5905, 1857594, 1146.4])
tb.add_row(["Darwin", 112, 120900, 1714.7])
tb.add_row(["Hobart", 1357, 205556,619.5])
tb.add_column('index',[1,2,3,4])   #* 添加一列 *
print(tb)

输出:

+-----------+------+------------+-----------------+-------+
| City name | Area | Population | Annual Rainfall | index |
+-----------+------+------------+-----------------+-------+
|  Adelaide | 1295 |  1158259   |      600.5      |   1   |
|  Brisbane | 5905 |  1857594   |      1146.4     |   2   |
|   Darwin  | 112  |   120900   |      1714.7     |   3   |
|   Hobart  | 1357 |   205556   |      619.5      |   4   |
+-----------+------+------------+-----------------+-------+

3、从csv文件添加数据

PrettyTable 不仅提供了手动按行按列添加数据,也支持直接从 csv 文件中读取数据。

import prettytable as pt
from prettytable import from_csv 

table = pt.PrettyTable()
with open('res.csv', 'r') as f:
    table = from_csv(fp) 
    print(table)

注意:csv文件不能通过xls直接重命名得到,会报错。如果是xls文件,请用另存为csv获得csv文件

4、从sql查询值添加

从数据库查询出来的数据可以直接导入到表格打印,下面的例子使用了sqlite3,如果使用的是mysql也是一样的,只要能查询到数据就能导入到表格中

import sqlite3
from prettytable import from_db_cursor 

conn = sqlite3.connect("/tmp/sqlite.db")
cur = conn.cursor()
cur.execute("SELECT * FROM res") 
table = from_db_cursor(cur)
print(table)

5、从HTML导入数据

支持从html的表格中导入,请看下面这个例子:

from prettytable import from_html

html_string='''<table>
<tr>
<th>code</th>
<th>uuid</th>
<th>name</th>
<th>IP</th>
</tr>
<tr>
<td>1</td>
<td>server01</td>
<td>server-01</td>
<td>192.168.100.1</td>
</tr>
<tr>
<td>2</td>
<td>server02</td>
<td>server-02</td>
<td>192.168.100.2</td>
</tr>
</table>'''

table = from_html(html_string)

print(table[0])

导入 html 的表格,但是不一样的地方是 print 语句,使用 html 表格导入数据的时候 print 的必须是列表中的第一个元素,否则有可能会报[<prettytable.PrettyTable object at 0x7fa87feba590>]这样的错误。

四、选择性输出


prettytable 在创建表格之后,你依然可以有选择的输出某些特定的行。

## 输出指定的列
print(table.get_string(fields=["Area", "Population"]))

## 输出指定的行,start 和 end 参数可以自由控制显示区间
print(table.get_string(start = 0, end = 2))

## 将表格切片
new_table = table[0:2]
print(new_table)

## 输出排序
print(table.get_string(sortby="City name", reversesort=True))
## 其中 reversesort 指定了是否倒序排序,默认为 False,即默认正序列排序。sortby 指定了排序的字段。

五、表格的样式


通过 set_style() 可以设置表格样式

import prettytable as pt

tb = pt.PrettyTable()
tb.field_names = ["City name", "Area", "Population", "Annual Rainfall"]
tb.add_row(["Adelaide",1295, 1158259, 600.5])
tb.add_row(["Brisbane",5905, 1857594, 1146.4])
tb.add_row(["Darwin", 112, 120900, 1714.7])
tb.add_row(["Hobart", 1357, 205556,619.5])

tb.set_style(pt.MSWORD_FRIENDLY)

print(tb)

输出:

| City name | Area | Population | Annual Rainfall | index |
|  Adelaide | 1295 |  1158259   |      600.5      |   1   |
|  Brisbane | 5905 |  1857594   |      1146.4     |   2   |
|   Darwin  | 112  |   120900   |      1714.7     |   3   |
|   Hobart  | 1357 |   205556   |      619.5      |   4   |

1、设置对齐方式

align 提供了用户设置对齐的方式,值有 l,r,c 方便代表 左对齐,右对齐和居中 如果不设置,默认居中对齐。

import prettytable as pt

tb = pt.PrettyTable()
tb.field_names = ["City name", "Area", "Population", "Annual Rainfall"]
tb.add_row(["Adelaide",1295, 1158259, 600.5])
tb.add_row(["Brisbane",5905, 1857594, 1146.4])
tb.add_row(["Darwin", 112, 120900, 1714.7])
tb.add_row(["Hobart", 1357, 205556,619.5])

tb.align["City name"] = "l"
tb.align["Area"] = "c"
tb.align["Population"] = "r"
tb.align["Annual Rainfall"] = "c"

print(tb)

输出:

+-----------+------+------------+-----------------+
| City name | Area | Population | Annual Rainfall |
+-----------+------+------------+-----------------+
| Adelaide  | 1295 |    1158259 |      600.5      |
| Brisbane  | 5905 |    1857594 |      1146.4     |
| Darwin    | 112  |     120900 |      1714.7     |
| Hobart    | 1357 |     205556 |      619.5      |
+-----------+------+------------+-----------------+

2、控制边框样式

在 PrettyTable 中,边框由三个部分组成,横边框,竖边框,和边框连接符(横竖交叉的链接符号)

import prettytable as pt

tb = pt.PrettyTable()
tb.field_names = ["City name", "Area", "Population", "Annual Rainfall"]
tb.add_row(["Adelaide",1295, 1158259, 600.5])
tb.add_row(["Brisbane",5905, 1857594, 1146.4])
tb.add_row(["Darwin", 112, 120900, 1714.7])
tb.add_row(["Hobart", 1357, 205556,619.5])

tb.border = True
tb.junction_char='$'
tb.horizontal_char = '+'
tb.vertical_char = '%'

print(tb)

说明:

  • table.border:控制是否显示边框,默认是 True
  • table.junction_char:控制边框连接符
  • table.horizontal_char:控制横边框符号
  • table.vertical_char:控制竖边框符号

输出:

$+++++++++++$++++++$++++++++++++$+++++++++++++++++$
% City name % Area % Population % Annual Rainfall %
$+++++++++++$++++++$++++++++++++$+++++++++++++++++$
%  Adelaide % 1295 %  1158259   %      600.5      %
%  Brisbane % 5905 %  1857594   %      1146.4     %
%   Darwin  % 112  %   120900   %      1714.7     %
%   Hobart  % 1357 %   205556   %      619.5      %
$+++++++++++$++++++$++++++++++++$+++++++++++++++++$

参考

推荐阅读更多精彩内容

  • ORA-00001: 违反唯一约束条件 (.) 错误说明:当在唯一索引所对应的列上键入重复值时,会触发此异常。 O...
    我想起个好名字阅读 4,389评论 0 9
  • 1、通过CocoaPods安装项目名称项目信息 AFNetworking网络请求组件 FMDB本地数据库组件 SD...
    爱运动爱学习阅读 15,638评论 3 114
  • 本周各位同学都非常认真地完成对主题回答。进行总结过程中,如有不错误的地方,请及时告诉我,在每个主题中增加了我出题最...
    oseca阅读 126评论 0 1
  • 今天下午奔驰S500着车,但是不稳。后面发现邮箱进谁了,今天把汽油抽出来,明天抬邮箱清洗
    京心达王博阅读 69评论 0 0
  • 有彩虹出现的晴天 谁给的思念 像窗边的飞燕 告诉我她的体贴 云舒云卷一片一片 想象她还在身边 抬头偷偷看一面 感觉...
    徐秀美阅读 173评论 0 0