input()语句不能放入函数中吗？我看3个功能函数都没放进去

if c == '1':
   n = input("您想要生成几注双色球：")
   generate_unionlotto(n)

elif c == '2':
   n = input("请输入您要查询的机构或者电话号码：")
   find_phone(keyword = n)

elif c == '3':
   n = input("请输入您要查询的城市：")
   w = get_weather(n)
   #print(w)
   if "name" in w:
       print("{date}   {name}  {weather}   {max}/{min} {wind}".format_map(w))
   else:
       print("未找到{0}的天气数据".format(n))

就这部分内容直接也写入函数中可以吗