从函数列表返回函数的python函数

Question

我要执行以下功能：

 1)input is a number. 
 2)functions are indexed, return a function whose index matches given number

这是我想出的：

def foo_selector(whatfoo):
    def foo1():
        return
    def foo2():
        return
    def foo3():
        return
    ...

    def foo999():
        return

    #something like return foo[whatfoo]

问题是，如何索引函数（foo＃）？ 我可以通过dir（）看到函数foo1到foo999。 但是，dir（）返回此类函数的名称，而不是函数本身。 在示例中，这些foo函数没有执行任何操作。 但是，在我的程序中，它们执行不同的任务，因此我无法自动生成它们。 我自己写它们，必须按他们的名字归还它们。

Answer 1

使用装饰器累积功能列表。

func_list = []

def listed_func(func):
    func_list.append(func)
    return func

@listed_func
def foo1():
   pass

@listed_func
def foo2():
   pass

现在，您可以通过列表中的索引轻松访问功能。

如果要按名称访问函数，也可以创建字典：

func_dict = {}

def collected_func(func):
    func_dict[func.__name__] = func
    return func

或从名称中提取索引，并将其用作dict键（由于dict不排序，因此，如果要按某种顺序遍历它们，则需要对键进行排序）：

func_dict = {}

def collected_func(func):
    key = int("".join(c for c in func.__name__ if c.isdigit()))
    func_dict[key] = func
    return func

或者显式地将索引号传递给装饰器：

func_dict = {}

def collected_func(key):
    def decorator(func):
        func_dict[key] = func
        return func
    return decorator

@collected_func(12)
def foo():
    pass

Answer 2

您可以将所有函数简单地放入一个数组中，例如：

def foo_selector(whatfoo):
    def foo1():
        return
    def foo2():
        return
    def foo3():
        return
    ...

    def foo999():
        return

    foo_list = [
        foo1,
        foo2,
        ...
        foo999
    ]

    return foo_list[whatfoo]

Answer 3

您可以通过以下几种方法来做到这一点：

eval("foo{}()".format(whatfoo))

要么

locals()['foo{}'.format(whatfoo)]