我正在尝试编写一个包含一些数学函数的小型Python模块。例如,它可能包含如下函数:
def quad(x, a, b, c):
return a*x**2 + b*x + c
您可能会注意到它除了变量a, b, c
之外还包含几个参数(即x
)。现在,如果我将它放在一个文件中并简单地导入它,最终用户必须始终使用除变量之外的参数调用该函数。因此我想创建一个这样的类:
class quad:
def __init__(self, a, b, c):
self.a = a
self.b = b
self.c = c
def eq(x):
return self.a*x**2 + self.b*x + self.c
因此允许最终用户将其用作:
q = quad(p, q, r)
eq = q.eq
这是正确的做事方式吗?我对这个问题的标题非常抱歉,因为我想不出更好的一个!
答案 0 :(得分:7)
这似乎是一个完全合理的课程用法。基本上,当程序涉及可以建模为具有状态的对象的事物时,您应该考虑使用类。此处,多项式的“状态”只是系数a
,b
和c
。
你也可以使用Python的__call__
特殊方法让你像对待函数一样对待类:
class quad:
def __init__(self, a, b, c):
self.a = a
self.b = b
self.c = c
def __call__(x):
return self.a * x**2 + self.b * x + self.c
q = quad(p, q, r)
q(x)
另一种方法,它可以稍微清洁,只是返回一个函数,其中系数被烘焙。正如Tichodrama所提到的那样,这基本上是一个currying的例子:
def quad(a, b, c):
def __quad(x):
return a * x**2 + b * x + c
return __quad
或使用lambda
语法:
def quad(a, b, c):
return lambda x: a * x**2 + b * x + c
然后可以这样使用:
q = quad(p, q, r)
q(x)
答案 1 :(得分:5)
看起来您正在搜索类似currying的内容。
也许this question可以帮到你。