Python计算水费

Python计算水费，请问是怎么算的，没有思路，希望可以指点一下

望采纳！！点击该回答右侧的“采纳”按钮即可采纳！

def calculate_water_fee(water_usage, family_size):
  # 计算阶梯水费
  base_water_usage = 220 + (family_size - 3) * 48
  if water_usage <= base_water_usage:
    water_fee = water_usage * 3.45
  elif water_usage <= base_water_usage + 80:
    water_fee = base_water_usage * 3.45 + (water_usage - base_water_usage) * 4.83
  else:
    water_fee = base_water_usage * 3.45 + 80 * 4.83 + (water_usage - base_water_usage - 80) * 5.83
  return water_fee

# 读取用户输入的用水量和家庭人数
water_usage = float(input("请输入用水量(立方米)："))
family_size = int(input("请输入家庭人数："))

# 计算水费
water_fee = calculate_water_fee(water_usage, family_size)

# 输出水费
print("应缴纳的水费为：", water_fee, "元")

公式都有了～剩下的不就简单了