编辑代码

def main():
   # 提示用户输入车辆总功率
   total_power = input("请输入车辆总功率： ")
   # 显示用户输入的车辆总功率
   print(f"您输入的车辆总功率是： {total_power} kW")

   # 提示用户输入发动机功率
   engine_power = input("请输入发动机功率： ")
   # 显示用户输入的发动机功率
   print(f"您输入的发动机功率是： {engine_power} kW")

   # 提示用户输入电机功率
   motor_power = input("请输入电机功率： ")
   # 显示用户输入的电机功率
   print(f"您输入的电机功率是： {motor_power} kW")

   # 提示用户输入电池组模块成本价格
   battery_cost = input("请输入电池组模块成本价格： ")
   # 显示用户输入的电池组模块成本价格
   print(f"您输入的电池组模块成本价格是： {battery_cost} 元")

   # 提示用户输入车辆是否为增程式混合动力汽车
   is_extended_range = input("车辆是否为增程式混合动力汽车 (是/否)： ")
   # 显示用户输入的车辆是否为增程式混合动力汽车
   print(f"车辆是否为增程式混合动力汽车： {is_extended_range}")

   # 根据用户输入的是否为增程式混合动力汽车，计算成本减少价格
   if is_extended_range.lower() == "是":
      # 计算成本减少价格
      cost_reduction = 588.60 + (int(engine_power) * 19.53) + (((int(motor_power) - int(engine_power)) / int(motor_power)) * int(battery_cost))
      print(f"成本减少价格为： {cost_reduction} 元")
   elif is_extended_range.lower() == "否":
      print("成本减少价格为588.60元")
   else:
      print("输入无效，请输入'是'或'否'来回答是否为增程式混合动力汽车")

if __name__ == "__main__":
   main()