在 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