Python simulation of high-speed rail ticketing system-using prettytable library

import prettytable as pt


# 显示坐席
def show_tickets(row_num):
    tb = pt.PrettyTable()
    tb.field_names = ["行号", "1", "2", "3", "4", "5"]
    for i in range(row_num):
        lst = [f"第{i + 1}排", "有票", "有票", "有票", "有票", "有票"]
        tb.add_row(lst)
    print(tb)


# 订票
def order_tickets(row_num, row, column):
    tb = pt.PrettyTable()
    tb.field_names = ["行号", "1", "2", "3", "4", "5"]
    for i in range(row_num):
        if i + 1 == int(row):
            lst = [f"第{i + 1}排", "有票", "有票", "有票", "有票", "有票"]
            lst[int(column)] = "已售"
            tb.add_row(lst)
        else:
            lst = [f"第{i + 1}排", "有票", "有票", "有票", "有票", "有票"]
            tb.add_row(lst)
    print(tb)


if __name__ == '__main__':
    row_num = 13
    show_tickets(row_num)
    while 1:
        choose_ticket = input("请输入选择的座位（请使用英文逗号隔开，如13排1座：13,1）:")
        try:
            row, column = choose_ticket.split(",")
            break
        except Exception:
            print("输入有误。请重新输入！")
    order_tickets(row_num, row, column)