用NumPy实现一个电影院选座程序

要求1：电影院售票代码

1：生成一个座位数组setArr,尺寸10行10列，元素为0

2：将卖出去的座位值置1 （例如：4排3列的座位），如果是已经卖出去的座位，请提示“该座位已经售出”;可以进行循环售票

3：将座位售票状态保存到本地set.txt(格式为整数，逗号分隔)

# 代码如下

import numpy as np
#1.生成一个座位数组setArr，大小为10行10列，元素是全0.
setArr = np.zeros((10,10),dtype=np.int32)
print(setArr)
#2.将卖出去的座位值置1（例如：4排3列的座位），如果是已经卖出去的座位，请提示“该座位已经售出”，可以进行循环售票。
while True:
    print("行号输入-1时退出程序")
    input_set_row = int(input("请输入座位行号"))
    if input_set_row == -1:
        break
    input_set_col = int(input("请输入座位列号"))
    if setArr[input_set_row-1,input_set_col-1]:
        print("座位已经售出，请重新选座")
    else:
        setArr[input_set_row-1,input_set_col-1]=1
    print(setArr)
#3.将座位售票状态保存到本地set.txt（格式为整数，分隔符为逗号）
np.savetxt("set.txt",setArr,fmt = "%d",delimiter=",")

注意：以上代码可以变换为“动车选座，飞机选座”等。有一个问题，大家可以探讨，单座位所剩无几时，怎么随机分配座位。