问题描述
我目前正在 Python 的 GUI 应用程序中构建“公式编辑器”。
我使用 sympy 根据 subs 字典来解析和检查输入公式的有效性。没问题。
有时，用户需要输入带有冗余参数的“复杂”公式。 请参阅下面的示例：
Array([0.9,0.8,1.0,1.1])[Mod(p-1,4)]
列表 [0.9,1.1] 由用户选择并且可以是任意长度。给定 p 变量的值，il 将产生列表的四个元素之一。数字 4 显然是 len([0.9,1.1]。
用户很容易输错公式...
相反，我想创建自己的函数，例如。 userlist()，将列表作为参数并根据需要进行操作。
我已经阅读了 this，它帮助我从以数字为参数的函数开始。它对列表参数没有多大帮助。
提前致谢。
编辑：
简而言之，我需要以某种方式定义 userlist() 以便这一行
parse_expr("userlist(p,[8,4,6,7])").evalf(subs={'p': 10})
返回列表的第 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,7])",{'userlist': lambda a,b: print(a,b)})
p [8,7]
此处的 lambda 函数可以是您想要使用的任何 Python 函数，它将接收解析后的列表作为参数，以便它可以调用 len 等。