python之prettytable模塊
python的prettytable模塊可以讓我們更加清晰的將數據記錄格式化打印出來,今天簡單看下這個模塊的使用方法。
注意:使用前需要引入包
from prettytable import PrettyTable
from prettytable import from_csv
如果對應的python包不存在,則需要通過:
pip install prettytable
來進行安裝。
方法一、直接添加列名字和數據記錄
from prettytable import PrettyTable
import sys
reload(sys)
sys.setdefaultencoding('utf-8')
# 直接輸入行和列進行排列
table = PrettyTable()
table.field_names = ["姓名", "年齡", "性別", "身高"]
table.add_row(["yeyz", 26, '男', 182])
table.add_row(["任盈盈", 27, '女', 172])
table.add_row(["楊過", 28, '男', 175])
# 添加一個列
table.add_column("成績", [60, 70, 80])
print table
輸出結果如下:
+--------+------+------+------+----+
| 姓名 | 年齡 | 性別 | 身高 | 成績 |
+--------+------+------+------+----+
| yeyz | 26 | 男 | 182 | 60 |
| 任盈盈 | 27 | 女 | 172 | 70 |
| 楊過 | 28 | 男 | 175 | 80 |
+--------+------+------+------+----+
Process finished with exit code 0
可以看到:
通過field_names可以將列名稱給列出來
通過add_row函數可以將記錄添加到表格里面
通過add_column可以給這個表添加數據記錄
方法二、通過加載csv文件來格式化輸出數據
from prettytable import PrettyTable
from prettytable import from_csv
import sys
reload(sys)
sys.setdefaultencoding('utf-8')
# 通過一個文件進行添加
'''
+--------+------+------+------+------+
| 姓名 | 年齡 | 性別 | 身高 | 分數 |
+--------+------+------+------+------+
| 劉德華 | 56 | 男 | 165 | 30 |
| 張學友 | 50 | 男 | 65 | 80 |
| 周傑倫 | 38 | 男 | 170 | 90 |
| 黎明 | 40 | 男 | 180 | 100 |
+--------+------+------+------+------+
'''
table1 = PrettyTable()
csv_file = open('csv_file.csv', 'r')
print csv_file
table1 = from_csv(csv_file)
print table1
輸出結果如下:
+--------+------+------+------+------+
| 姓名 | 年齡 | 性別 | 身高 | 分數 |
+--------+------+------+------+------+
| 劉德華 | 56 | 男 | 165 | 30 |
| 張學友 | 50 | 男 | 65 | 80 |
| 周傑倫 | 38 | 男 | 170 | 90 |
| 黎明 | 40 | 男 | 180 | 100 |
+--------+------+------+------+------+