时间戳

首页 > 解决方案 > 是否有更 Pythonic 的方式通过字符串名称访问函数？

python - 是否有更 Pythonic 的方式通过字符串名称访问函数？

问题描述

我有一些函数（硬编码，我不希望 Python 计算导数，实际上这些函数更复杂）：

def f(x): return x
def g(x): return x ** 2
def f_prime(x): return 1
def g_prime(x): return 2 * x

和一个函数列表：

myfunctions = [f, f, g, f]

如何获取相关“主要”功能的列表？

我在想：

myderivatives = [globals()[function.__name__ + '_prime'] for function in myfunctions]

它有效：

[<function f_prime at 0x00000000022CF048>, <function f_prime at 0x00000000022CF048>, <function g_prime at 0x00000000022CF0B8>, <function f_prime at 0x00000000022CF048>]

但是有没有一种更 Pythonic 的方式来获取与原始版本关联的“主要”函数列表？

还有一个角落案例：

from module1 import thisfunction as f
def f_prime(x): 
    pass

globals()[f.__name__ + '_prime']  
# doesn't work because f.__name__ + '_prime' is 'thisfunction_prime' and not 'f_prime'

标签： pythonfunctionvariables

解决方案

定义一个字典来保存你的函数，然后使用列表推导：

funcd = {'f': f, 'g': g, 'f_prime': f_prime, 'g_prime': g_prime}

myderivatives = [func for name, func in funcd.items() if name.endswith('_prime')]

此解决方案的好处是它精确地定义了您希望过滤的功能范围。不推荐使用globals，也很少需要使用。

推荐阅读