Python中的prettytable模块

Python中的prettytable模块

Python中的prettytable模块是一个用来创建并打印美观的表格的工具。在数据分析、数据可视化以及文本输出方面，prettytable模块都有着重要的作用。在本文中，我们将详细介绍prettytable模块的使用方法和功能。

1. 安装prettytable模块

要使用prettytable模块，首先需要安装它。可以使用pip命令来安装prettytable模块：

pip install prettytable

2. 创建表格

使用prettytable模块创建表格非常简单。首先需要导入prettytable模块，然后创建一个PrettyTable对象，并添加数据到表格中。下面是一个简单的示例：

from prettytable import PrettyTable

# 创建一个表格对象
table = PrettyTable()

# 添加表头
table.field_names = ["姓名", "年龄", "性别"]

# 添加数据行
table.add_row(["张三", 25, "男"])
table.add_row(["李四", 30, "女"])

# 打印表格
print(table)

运行上面的代码，将输出如下的表格：

+--------+--------+--------+
|  姓名  |  年龄  |  性别  |
+--------+--------+--------+
|  张三  |  25   |  男   |
|  李四  |  30   |  女   |
+--------+--------+--------+

3. 设置表格样式

prettytable模块还支持设置表格的各种样式，包括对齐方式、边框线样式、表头样式等。下面是一个示例：

# 设置对齐方式
table.align["姓名"] = "l"
table.align["年龄"] = "r"
table.align["性别"] = "c"

# 设置边框线样式
table.hrules = PrettyTable.ALL

# 设置表头样式
table.field_names = [f'\033[1;31m{x}\033[0m' for x in table.field_names]

print(table)

运行上面的代码，将输出如下的表格，其中姓名为红色加粗：

+--------+--------+--------+
|  姓名  |  年龄  |  性别  |
+--------+--------+--------+
|  张三  |      25 |   男   |
|  李四  |      30 |   女   |
+--------+--------+--------+

4. 添加数据

除了手动添加数据外，prettytable模块还支持从列表、字典等数据结构中批量添加数据到表格中。下面是一个示例：

# 从列表中添加数据
data = [
    ["王五", 28, "男"],
    ["赵六", 35, "女"]
]

table.add_rows(data)

print(table)

运行上面的代码，将输出包含全部数据的表格：

+--------+--------+--------+
|  姓名  |  年龄  |  性别  |
+--------+--------+--------+
|  张三  |      25 |   男   |
|  李四  |      30 |   女   |
|  王五  |      28 |   男   |
|  赵六  |      35 |   女   |
+--------+--------+--------+

5. 导出表格

prettytable模块还支持将表格导出为不同的格式，如文本、CSV、HTML等。下面是一个示例：

# 导出为文本
with open("table.txt", "w") as file:
    file.write(str(table))

# 导出为CSV
table.to_csv("table.csv")

# 导出为HTML
table_str = table.get_html_string()
with open("table.html", "w") as file:
    file.write(table_str)

运行上面的代码，将在当前目录下生成table.txt、table.csv和table.html文件，分别对应文本、CSV和HTML格式的表格导出。

结论

通过本文的介绍，我们详细了解了Python中的prettytable模块的基本用法和功能。prettytable模块可以帮助我们轻松创建并打印美观的表格，方便展示和分析数据。

本文链接：http://so.lmcjl.com/news/4689/