python字符串切片问题

Question

题目：已知字符串s='苹果:12个;梨:20个;桔子:35个;香蕉:18个'，试编写程序，实现以下功能：

1. 根据输入的水果，查询个数。（假设用户输入的水果在字符串中均存在）
   例如：输入“苹果”，输出“苹果总共有12个。”

我的答案：
str='苹果:12个;梨:20个;桔子:35个;香蕉:18个'
a=int(input("请输入水果:"))
b=(a-1)*3
c=str[b:b+6]
print("水果共有",c)

运行的时候报错了，请问如何修改

Answer 1

# 定义字符串s
s = '苹果:12个;梨:20个;桔子:35个;香蕉:18个'

# 将字符串s按照分号分割，得到一个列表
lst = s.split(';')

# 定义一个空字典，用于存储水果和个数的对应关系
dic = {}

# 遍历列表中的每个元素
for item in lst:
  # 将每个元素按照冒号分割，得到水果和个数
  fruit, num = item.split(':')
  # 去掉个数中的'个'字，转换为整数类型
  num = int(num[:-1])
  # 将水果和个数作为键值对存入字典中
  dic[fruit] = num

# 提示用户输入水果
fruit = input('请输入水果：')

# 判断用户输入的水果是否在字典中
if fruit in dic:
  # 如果在字典中，输出对应的个数
  print(f'{fruit}总共有{dic[fruit]}个。')
else:
  # 如果不在字典中，输出提示信息
  print('没有找到该水果。')