写一个二手车脚本

以下是一个简单的二手车脚本示例：

import csv

def load_data(file_name):
    data = []
    with open(file_name, 'r') as file:
        reader = csv.reader(file)
        for row in reader:
            data.append(row)
    return data

def save_data(file_name, data):
    with open(file_name, 'w', newline='') as file:
        writer = csv.writer(file)
        writer.writerows(data)

def add_car(data):
    car = []
    car.append(input("请输入车辆品牌："))
    car.append(input("请输入车辆型号："))
    car.append(input("请输入车辆年份："))
    car.append(input("请输入车辆里程数："))
    car.append(input("请输入车辆价格："))
    data.append(car)
    print("车辆信息已添加成功！")

def delete_car(data):
    index = int(input("请输入要删除的车辆索引："))
    if index < len(data):
        car = data.pop(index)
        print(f"已成功删除车辆：{car[0]} {car[1]}")
    else:
        print("无效的车辆索引！")

def search_car(data):
    keyword = input("请输入要搜索的关键词：")
    found_cars = []
    for car in data:
        if keyword.lower() in ' '.join(car).lower():
            found_cars.append(car)
    print("搜索结果如下：")
    if found_cars:
        for car in found_cars:
            print(f"{car[0]} {car[1]}")
    else:
        print("未找到相关车辆！")

def display_menu():
    print("\n欢迎使用二手车管理系统！")
    print("1. 添加车辆信息")
    print("2. 删除车辆信息")
    print("3. 搜索车辆信息")
    print("4. 显示所有车辆信息")
    print("5. 退出系统")

def main():
    data = load_data("cars.csv")
    while True:
        display_menu()
        choice = input("请输入您的选择：")
        if choice == '1':
            add_car(data)
        elif choice == '2':
            delete_car(data)
        elif choice == '3':
            search_car(data)
        elif choice == '4':
            for car in data:
                print(f"{car[0]} {car[1]} {car[2]} {car[3]} {car[4]}")
        elif choice == '5':
            save_data("cars.csv", data)
            print("谢谢使用，再见！")
            break
        else:
            print("无效的选择！")

if __name__ == "__main__":
    main()

这个脚本使用了CSV文件来存储二手车的信息。它提供了添加车辆信息、删除车辆信息、搜索车辆信息和显示所有车辆信息的功能。用户可以根据菜单进行操作，并在退出系统时将数据保存回CSV文件中