在 SymPy 中定义一个将列表作为参数的函数
Defining a function in SymPy that takes a list as argument
我目前正在 python 的 GUI 应用程序中构建“公式编辑器”。
我使用 sympy 来解析和检查键入公式的有效性,基于 subs
的字典。没问题。
有时,用户需要键入带有冗余参数的“复杂”公式。
请参阅下面的示例:
Array([0.9, 0.8, 1.0, 1.1])[Mod(p-1, 4)]
列表[0.9, 0.8, 1.0, 1.1]
由用户选择,长度不限。给定 p
变量的值,il 将产生列表的四个元素之一。数字4
显然是len([0.9, 0.8, 1.0, 1.1]
。
用户很容易输入错误的公式...
相反,我想创建自己的函数,例如。 userlist()
,将列表作为参数并按需运行。
我已阅读 ,它帮助我开始使用以数字作为参数的函数。它对列表参数没有多大帮助。
提前致谢。
编辑:
简而言之,我需要以某种方式定义 userlist()
以便这一行
parse_expr("userlist(p, [8, 4, 6, 7])").evalf(subs={'p': 10})
returns 列表的第 Mod(p-1, len(list))
个元素(这里是第二个元素:4
)。
我不确定我是否理解您的问题,但也许这会有所帮助:
In [75]: A = IndexedBase('A')
In [76]: A[Mod(p-1, 4)]
Out[76]: A[Mod(p + 3, 4)]
In [77]: f = lambdify((p, A), A[Mod(p-1, 4)])
In [78]: f(2, [3, 5, 6, 7])
Out[78]: 5
好的,我看到你更新的问题了。我仍然不确定我是否完全理解,但我希望这个示例会更有帮助:
In [7]: parse_expr("userlist(p, [8, 4, 6, 7])", {'userlist': lambda a, b: print(a, b)})
p [8, 4, 6, 7]
此处的 lambda
函数可以是您要使用的任何 Python 函数,它将接收已解析的列表作为参数,因此它可以调用 len
等
我目前正在 python 的 GUI 应用程序中构建“公式编辑器”。
我使用 sympy 来解析和检查键入公式的有效性,基于 subs
的字典。没问题。
有时,用户需要键入带有冗余参数的“复杂”公式。 请参阅下面的示例:
Array([0.9, 0.8, 1.0, 1.1])[Mod(p-1, 4)]
列表[0.9, 0.8, 1.0, 1.1]
由用户选择,长度不限。给定 p
变量的值,il 将产生列表的四个元素之一。数字4
显然是len([0.9, 0.8, 1.0, 1.1]
。
用户很容易输入错误的公式...
相反,我想创建自己的函数,例如。 userlist()
,将列表作为参数并按需运行。
我已阅读
提前致谢。
编辑:
简而言之,我需要以某种方式定义 userlist()
以便这一行
parse_expr("userlist(p, [8, 4, 6, 7])").evalf(subs={'p': 10})
returns 列表的第 Mod(p-1, len(list))
个元素(这里是第二个元素:4
)。
我不确定我是否理解您的问题,但也许这会有所帮助:
In [75]: A = IndexedBase('A')
In [76]: A[Mod(p-1, 4)]
Out[76]: A[Mod(p + 3, 4)]
In [77]: f = lambdify((p, A), A[Mod(p-1, 4)])
In [78]: f(2, [3, 5, 6, 7])
Out[78]: 5
好的,我看到你更新的问题了。我仍然不确定我是否完全理解,但我希望这个示例会更有帮助:
In [7]: parse_expr("userlist(p, [8, 4, 6, 7])", {'userlist': lambda a, b: print(a, b)})
p [8, 4, 6, 7]
此处的 lambda
函数可以是您要使用的任何 Python 函数,它将接收已解析的列表作为参数,因此它可以调用 len
等