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")

到目前为止，我已经在文档字符串中添加了“提示”，但是我想要这样的东西：

它可能显示input_var ='年/月/日'或类似的内容。

使用带有Python 3的Jupyter笔记本

谢谢

解决方法

• 查看功能注释。
• 示例：

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

• 现在，如果您键入x并按Tab键以获取建议，它将显示x : 'year/month/day'。