python实操案例__02—利用prettytable库模拟高铁售票系统

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库的基本用法有了更深入的了解。

通过表格的形式,我们可以方便地显示和操作数据。使用条件语句,我们可以根据用户输入做出不同的操作,实现了简单的售票系统功能。

值得注意的是,本案例中的数据和操作仅为示范目的,实际中可能需要更多的功能和数据处理。

后端开发标签