Jupyter中函数输入的Python提示

是否可以添加某种装饰器或别名来获取变量在函数中可能获得的输入提示

在下面的示例中，函数input_var可以接受任何值，但我是否可以提示使用值'year'、'month'或'day'

def some_function(input_var):
    """ This function does something cool, use input_var as bla, blah or blahh """
    if input_var=='year':
        print("Selected 1")
    if input_var=='month':
        print("Selected 2")
    if input_var=='day':
        print("Selected 3")

到目前为止，我将“提示”添加到docstring中，但我想要如下内容：

它可以显示输入值_var='year/month/day'或类似的内容

使用Jupyter笔记本电脑，使用python 3

谢谢

• 看看函数注释
• 例如：

• 现在，如果您键入

  x

  并按tab键获取建议，它将显示

  x:“年/月/日”

---

请看这里可能有帮助的函数注释。根据PEP-563，对类型提示以外的内容使用函数注释有些不推荐。

def foo(x: 'year/month/day'):
  return "abc"