python实操案例__02—利用prettytable库模拟高铁售票系统
在本文中,我们将使用Python编写一个模拟高铁售票系统的案例。我们将使用一个名为prettytable的库来创建漂亮的表格,并使用用户输入和条件语句来模拟售票系统的功能。在这个案例中,我们将介绍如何安装prettytable库、创建表格、添加数据以及根据用户输入做出相应的操作。
1. 安装prettytable库
首先,我们需要安装prettytable库,可以使用以下命令在命令行中安装:
pip install prettytable
成功安装后,我们就可以在Python脚本中引入prettytable库。
2. 创建表格
在本案例中,我们将创建一个表格来存储高铁的车次、出发地、目的地、日期和剩余座位数。
from prettytable import PrettyTable
# 创建表格对象
table = PrettyTable()
# 设置表头
table.field_names = ["车次", "出发地", "目的地", "日期", "剩余座位数"]
我们使用PrettyTable类创建了一个名为table的表格对象,并通过field_names属性设置了表头,即表格的列名。
3. 添加数据
假设我们有以下几个高铁车次:
trains = [
{"车次": "G1234", "出发地": "北京", "目的地": "上海", "日期": "2022-01-01", "剩余座位数": 100},
{"车次": "G5678", "出发地": "上海", "目的地": "北京", "日期": "2022-01-02", "剩余座位数": 50},
{"车次": "G9012", "出发地": "广州", "目的地": "深圳", "日期": "2022-01-03", "剩余座位数": 80},
]
我们可以使用for循环将这些车次信息添加到表格中:
for train in trains:
table.add_row(train.values())
在每次循环中,我们使用table对象的add_row方法,将每个车次的值添加到表格的一行中。
4. 显示表格
使用table对象的get_string方法可以将表格以字符串的形式显示出来:
print(table.get_string())
运行以上代码,我们可以在控制台上看到漂亮的表格显示出来。
5. 模拟售票系统
现在,我们将通过用户输入来实现一个简单的售票系统。用户可以输入车次,程序将根据车次查询并显示该车次的剩余座位数。
# 获取用户输入的车次
train_number = input("请输入要查询的车次:")
# 根据用户输入的车次查询并显示剩余座位数
for train in trains:
if train["车次"] == train_number:
print(f"车次:{train_number},剩余座位数:{train['剩余座位数']}")
在以上代码中,我们使用input函数获取用户输入的车次,并使用条件语句遍历trains列表,查找与用户输入相符的车次信息。如果找到匹配的车次,我们使用print函数显示该车次的剩余座位数。
6. 总结
在本文中,我们使用prettytable库创建了一个模拟高铁售票系统的案例。我们学习了如何安装prettytable库、创建表格、添加数据以及根据用户输入做出相应的操作。通过这个案例,我们对prettytable库的基本用法有了更深入的了解。
通过表格的形式,我们可以方便地显示和操作数据。使用条件语句,我们可以根据用户输入做出不同的操作,实现了简单的售票系统功能。
值得注意的是,本案例中的数据和操作仅为示范目的,实际中可能需要更多的功能和数据处理。