太漂亮了 ! 输出好看的表格，就用这个 Python 库！

时间：2022-09-04 17:49:25

太漂亮了 ! 输出好看的表格，就用这个 Python 库！

 # 1. 前言

最近在用 Python 写一个小工具，这个工具主要就是用来管理各种资源的信息，比如阿里云的 ECS 等信息，因为我工作的电脑使用的是 LINUX，所以就想着用 python 写一个命令行的管理工具，基本的功能就是同步阿里云的资源的信息到数据库，然后可以使用命令行查询。

因为信息是展现在命令行中的，众所周知，命令行展现复杂的文本看起来着实累人，于是就想着能像表格那样展示，那看起来就舒服多了。

prettytable 库就是这么一个工具，prettytable 可以打印出美观的表格，并且对中文支持相当好（如果有试图自己实现打印表格，你就应该知道处理中文是多么的麻烦）

# 2. 安装

prettytable 并非 python 的内置库，通过 pip install prettytable即可安装。

# 3. 示例

我们先来看一个示例：

  1. #!/usr/bin/python  
  2. #**coding:utf-8**  
  3. import sys  
  4. from prettytable import PrettyTable  
  5. reload(sys)  
  6. sys.setdefaultencoding('utf8')  
  7. table = PrettyTable(['编号','云编号','名称','IP地址'])  
  8. table.add_row(['1','server01','服务器01','172.16.0.1'])  
  9. table.add_row(['2','server02','服务器02','172.16.0.2'])  
  10. table.add_row(['3','server03','服务器03','172.16.0.3'])  
  11. table.add_row(['4','server04','服务器04','172.16.0.4'])  
  12. table.add_row(['5','server05','服务器05','172.16.0.5'])  
  13. table.add_row(['6','server06','服务器06','172.16.0.6'])  
  14. table.add_row(['7','server07','服务器07','172.16.0.7'])  
  15. table.add_row(['8','server08','服务器08','172.16.0.8'])  
  16. table.add_row(['9','server09','服务器09','172.16.0.9'])  
  17. print(table) 

以上示例运行结果如下：

  1. linuxops@deepin:~$ python p.py  
  2. +------+----------+----------+------------+  
  3. | 编号 |  云编号  |   名称   |   IP地址   |  
  4. +------+----------+----------+------------+  
  5. |  1   | server01 | 服务器01 | 172.16.0.1 |  
  6. |  2   | server02 | 服务器02 | 172.16.0.2 |  
  7. |  3   | server03 | 服务器03 | 172.16.0.3 |  
  8. |  4   | server04 | 服务器04 | 172.16.0.4 |  
  9. |  5   | server05 | 服务器05 | 172.16.0.5 |  
  10. |  6   | server06 | 服务器06 | 172.16.0.6 |  
  11. |  7   | server07 | 服务器07 | 172.16.0.7 |  
  12. |  8   | server08 | 服务器08 | 172.16.0.8 |  
  13. |  9   | server09 | 服务器09 | 172.16.0.9 |  
  14. +------+----------+----------+------------+ 

在以上的示例中，我们通过form导入了表格库。table实例化了一个表格库，并且添加了['编号','云编号','名称','IP地址']为表头，如果没有添加表头，那么会以默认的Field+编号显示，例如：

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

所以为更直观看出每一列的意义，还是要添加表头的。

# 4. 添加数据

prettytable提供了多种的添加数据的方式，最常用的应该就是按行按列添加数据了。

按行添加数据 table.add_row

在上面简单的示例中，我们就是按行添加数据的。

添加的数据必须要是列表的形式，而且数据的列表长度要和表头的长度一样。在实际的使用中，我们应该要关注到添加的数据是否和表头对应，这一点很重要。

按列添加数据 table.add_column()

看下面的示例：

  1. #!/usr/bin/python  
  2. #**coding:utf-8**  
  3. import sys  
  4. from prettytable import PrettyTable  
  5. reload(sys)  
  6. sys.setdefaultencoding('utf8')  
  7. table = PrettyTable()  
  8. table.add_column('项目', ['编号','云编号','名称','IP地址'])  
  9. table.add_column('值', ['1','server01','服务器01','172.16.0.1'])  
  10. print(table) 

运行结果如下：

  1. +-------+--------+------------+  
  2. | index | 项目 |    值     |  
  3. +-------+--------+------------+  
  4. |   1   |  编号  |     1      |  
  5. |   2   | 云编号 |  server01  | 
  6. |   3   |  名称  |  服务器01   |  
  7. |   4   | IP地址 | 172.16.0.1 |  
  8. +-------+--------+------------+ 

以上示例中，我们通过add_column来按列添加数据，按列添加数据不需要在实例化表格的时候制定表头，它的表头是在添加列的时候指定的。

table.add_column('项目', ['编号','云编号','名称','IP地址']) 这一行代码为例，项目指定了这个列的表头名为"项目"，['编号','云编号','名称','IP地址']为列的值，同样为列表。

 从csv文件添加数据

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

  1. #!/usr/bin/python  
  2. #**coding:utf-8**  
  3. import sys  
  4. from prettytable import PrettyTable  
  5. from prettytable import from_csv  
  6. reload(sys)  
  7. sys.setdefaultencoding('utf8')  
  8. table = PrettyTable()  
  9. fp = open("res.csv", "r")  
  10. table = from_csv(fp)  
  11. print(table)  
  12. fp.close() 

如果要读取cvs文件数据，必须要先导入from_csv，否则无法运行。上面的示例运行结果如下：

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

 从sql查询值添加

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

  1. #!/usr/bin/python  
  2. #**coding:utf-8**  
  3. import sys  
  4. from prettytable import PrettyTable  
  5. from prettytable import from_db_cursor  
  6. import sqlite3  
  7. reload(sys)  
  8. sys.setdefaultencoding('utf8')  
  9. conn = sqlite3.connect("/tmp/aliyun.db")  
  10. cur = conn.cursor()  
  11. cur.execute("SELECT * FROM res")  
  12. table = from_db_cursor(cur)  
  13. print(table) 

运行结果如下：

  1. +------+----------+----------+------------+  
  2. | 编号 |  云编号  |   名称   |   IP地址   |  
  3. +------+----------+----------+------------+  
  4. |  1   | server01 | 服务器01 | 172.16.0.1 |  
  5. |  2   | server02 | 服务器02 | 172.16.0.2 |  
  6. |  3   | server03 | 服务器03 | 172.16.0.3 |  
  7. |  4   | server04 | 服务器04 | 172.16.0.4 |  
  8. |  5   | server05 | 服务器05 | 172.16.0.5 |  
  9. |  6   | server06 | 服务器06 | 172.16.0.6 |  
  10. |  7   | server07 | 服务器07 | 172.16.0.7 |  
  11. |  8   | server08 | 服务器08 | 172.16.0.8 |  
  12. |  9   | server09 | 服务器09 | 172.16.0.9 |  
  13. +------+----------+----------+------------+ 

从HTML导入数据

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

  1. #!/usr/bin/python  
  2. #**coding:utf-8**  
  3. import sys  
  4. from prettytable import PrettyTable  
  5. from prettytable import from_html  
  6. reload(sys)  
  7. sys.setdefaultencoding('utf8')  
  8. html_string='''<table>  
  9. <tr>  
  10. <th>编号</th>  
  11. <th>云编号</th>  
  12. <th>名称</th>  
  13. <th>IP地址</th>  
  14. </tr>  
  15. <tr>  
  16. <td>1</td>  
  17. <td>server01</td>  
  18. <td>服务器01</td>  
  19. <td>172.16.0.1</td>  
  20. </tr>  
  21. <tr>  
  22. <td>2</td>  
  23. <td>server02</td>  
  24. <td>服务器02</td>  
  25. <td>172.16.0.2</td>  
  26. </tr>  
  27. </table>'''  
  28. table = from_html(html_string)  
  29. print(table[0]) 

运行结果如下：

  1. +------+----------+----------+------------+  
  2. | 编号 |  云编号  |   名称   |   IP地址   |  
  3. +------+----------+----------+------------+ 
  4. |  1   | server01 | 服务器01 | 172.16.0.1 |  
  5. |  2   | server02 | 服务器02 | 172.16.0.2 |  
  6. +------+----------+----------+------------+ 

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

这是因为table并不是PrettyTable对象，而是包含单个PrettyTable对象的列表，它通过解析html而来，所以无法直接打印table，而需要打印table[0]

# 5. 表格输出格式

正如支持多种输入一样，表格的输出也支持多种格式，我们在上面中的例子中已经使用了print的方式输出，这是一种常用的输出方式。

 print

直接通过print打印出表格。这种方式打印出的表格会带边框。

 输出HTML格式的表格

print(table.get_html_string())可以打印出html标签的表格。

在上面的例子中，使用print(table.get_html_string())会打印出如下结果：

  1. <table>  
  2.     <tr>  
  3.         <th>编号</th>  
  4.         <th>云编号</th>  
  5.         <th>名称</th>  
  6.         <th>IP地址</th>  
  7.     </tr>  
  8.     <tr>  
  9.         <td>1</td>  
  10.         <td>server01</td>  
  11.         <td>服务器01</td>  
  12.         <td>172.16.0.1</td>  
  13.     </tr>  
  14.     <tr>  
  15.         <td>2</td>  
  16.         <td>server02</td>  
  17.         <td>服务器02</td>  
  18.         <td>172.16.0.2</td>  
  19.     </tr>  
  20. </table> 

# 6. 选择性输出

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

 输出指定的列

print table.get_string(fields=["编号", "IP地址"])可以输出指定的列

 输出前两行

通过print(table.get_string(start = 0, end = 2))的可以打印出指定的列，当然start和end参数让我可以*控制显示区间。当然区间中包含start不包含end，是不是很熟悉这样的用法？

根据输出指定行列的功能，我们可以同时指定行和列来输出，这里就不说明了。

 将表格切片

从上面的输出区间，我们做一个大胆的假设，既然区间包含start不包含end这种规则和切片的一样，我们可以不可通过切片来生成一个新的表格然后将其打印。

事实上是可以的。

  1. new_table = table[0:2]  
  2. print(new_table) 

如上代码段中，我们就可以打印出0到1行共2行的表格，python的切片功能异常强大，配合切片我们可以*的输入任意的行。

 输出排序

有时候我们需要对输出的表格进行排序，使用print table.get_string(sortby="编号", reversesort=True)就可以对表格进行排序，其中reversesort指定了是否倒序排序,默认为False，即默认正序列排序。

sortby指定了排序的字段。

# 7. 表格的样式

 内置样式

通过set_style()可以设置表格样式，prettytable内置了多种的样式个人觉得MSWORD_FRIENDLY，PLAIN_COLUMNS，DEFAULT 这三种样式看起来比较清爽，在终端下显示表格本来看起就很累，再加上一下花里胡哨的东西看起来就更累。

除了以上推荐的三种样式以外，还有一种样式不得不说，那就是RANDOM，这是一种随机的样式，每一次打印都会在内置的样式中随机选择一个，比较好玩。

具体内置了几种样式，请各位参考官网完整自己尝试输出看看。

  1. #!/usr/bin/python  
  2. #**coding:utf-8**  
  3. import sys  
  4. from prettytable import PrettyTable  
  5. from prettytable import MSWORD_FRIENDLY  
  6. from prettytable import PLAIN_COLUMNS  
  7. from prettytable import RANDOM  
  8. from prettytable import DEFAULT    
  9. reload(sys)  
  10. sys.setdefaultencoding('utf8')  
  11. table = PrettyTable(['编号','云编号','名称','IP地址'])  
  12. table.add_row(['1','server01','服务器01','172.16.0.1'])  
  13. table.add_row(['3','server03','服务器03','172.16.0.3'])  
  14. table.add_row(['2','server02','服务器02','172.16.0.2'])  
  15. table.add_row(['9','server09','服务器09','172.16.0.9'])  
  16. table.add_row(['4','server04','服务器04','172.16.0.4'])  
  17. table.add_row(['5','server05','服务器05','172.16.0.5'])  
  18. table.add_row(['6','server06','服务器06','172.16.0.6'])  
  19. table.add_row(['8','server08','服务器08','172.16.0.8'])  
  20. table.add_row(['7','server07','服务器07','172.16.0.7'])  
  21. table.set_style(DEFAULT)  
  22. print(table) 

 自定义样式

除了内置的样式以外，PrettyTable也提供了用户自定义，例如对齐方式，数字输出格式，边框连接符等等

 设置对齐方式

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

 控制边框样式

在PrettyTable中，边框由三个部分组成，横边框，竖边框，和边框连接符（横竖交叉的链接符号）

如下示例：

  1. #!/usr/bin/python  
  2. #**coding:utf-8**  
  3. import sys  
  4. from prettytable import PrettyTable  
  5. reload(sys)  
  6. sys.setdefaultencoding('utf8')  
  7. table = PrettyTable(['编号','云编号','名称','IP地址'])  
  8. table.add_row(['1','server01','服务器01','172.16.0.1'])  
  9. table.add_row(['3','server03','服务器03','172.16.0.3'])  
  10. table.add_row(['2','server02','服务器02','172.16.0.2'])  
  11. table.add_row(['9','server09','服务器09','172.16.0.9'])  
  12. table.add_row(['4','server04','服务器04','172.16.0.4'])  
  13. table.add_row(['5','server05','服务器05','172.16.0.5'])  
  14. table.add_row(['6','server06','服务器06','172.16.0.6'])  
  15. table.add_row(['8','server08','服务器08','172.16.0.8'])  
  16. table.add_row(['7','server07','服务器07','172.16.0.7'])  
  17. table.align[1] = 'l'  
  18. table.border = True  
  19. table.junction_char='$'  
  20. table.horizontal_char = '+'  
  21. table.vertical_char = '%'  
  22. print(table)  
  23. table.border`控制是否显示边框，默认是`True 

table.junction_char控制边框连接符

table.horizontal_char控制横边框符号

table.vertical_char控制竖边框符号

上例运行如下：

  1. $++++++$++++++++++$++++++++++$++++++++++++$  
  2. % 编号 %  云编号  %   名称   %   IP地址   %  
  3. $++++++$++++++++++$++++++++++$++++++++++++$  
  4. %  1   % server01 % 服务器01 % 172.16.0.1 %  
  5. %  3   % server03 % 服务器03 % 172.16.0.3 %  
  6. %  2   % server02 % 服务器02 % 172.16.0.2 %  
  7. %  9   % server09 % 服务器09 % 172.16.0.9 %  
  8. %  4   % server04 % 服务器04 % 172.16.0.4 %  
  9. %  5   % server05 % 服务器05 % 172.16.0.5 %  
  10. %  6   % server06 % 服务器06 % 172.16.0.6 %  
  11. %  8   % server08 % 服务器08 % 172.16.0.8 %  
  12. %  7   % server07 % 服务器07 % 172.16.0.7 %  
  13. $++++++$++++++++++$++++++++++$++++++++++++$ 

以上简单介绍了表格常用的一些样式设置。更具体的方法可参数请参考官方网站：

https://github.com/jazzband/prettytable

原文地址：https://mp.weixin.qq.com/s/O_X7N8TKFnapOWFarrXnEg

标签：

相关文章