python-模拟高铁售票系统

import prettytable as pt
# 没订票前
def show_ticket(row_num=6,columns_num=5):
    tb = pt.PrettyTable() #创建一张表格
    #设置列名
    colu_names = ["座位"+str(i) for i in range(1,columns_num + 1)]
    colu_names.insert(0, "行号")
    tb.field_names = colu_names
    data_num = len(colu_names) - 1
    # print(colu_names)
    #遍历初始化每一行
    for i in range(1,row_num + 1):
        lis = ["有票"] * data_num
        lis.insert(0, f"第{i}行")
        # print(lis)
        tb.add_row(lis)
    return tb

#订票
def order_tidcket(row, column, row_num=6, columns_num=5):
    tb = pt.PrettyTable()  # 创建一张表格
    # 设置列名
    colu_names = ["座位" + str(i) for i in range(1, columns_num + 1)]
    colu_names.insert(0, "行号")
    tb.field_names = colu_names
    data_num = len(colu_names) - 1
    #遍历初始化每一行
    for i in range(1, row_num + 1):
        lis = ["有票"] * data_num
        lis.insert(0, f"第{i}行")
        if int(row) == i:
            lis[int(column)] = "已售"
        tb.add_row(lis)
    return tb

#主函数
def main():
    print("没售票前！")
    print(show_ticket())
    print("开始售票！")
    site = input("请输入已售座位号(示例：如 3,4 表示3行4列)：")
    row, column = site.split(",")
    print("售票完成！")
    buyed = order_tidcket(row, column)
    print(buyed)

#程序入口
if __name__ == '__main__':
    main()

输出：