pattern-matching - SymPy 中所有可能匹配项的集合

Question

我想以返回所有可能匹配项的方式将模式与表达式匹配。使用 SymPy，我可以以一种任意选择的方式将我的模式与我的表达式匹配。我如何使用模式匹配来获取整个集合(直到任意重命名)

目前我可以做到这一点

>>> p, q = Wild('p'), Wild('q')
>>> x, y, z = symbols('x,y,z')

>>> (p+q).matches(x+y+z)
{p: y + z, q: x}

我想做这个

>>> (p+q).allmatches(x+y+z)
{{p: x, q: y + z}, {p: y, q: x + z}, {p: z, q: x + y}}

请注意，每个可能的分区都已表示。我愿意自己实现。产生所有可能匹配的模式匹配标准算法是什么？

Answer 1

我认为您必须使用所有可能的排除集来定位匹配项，例如

In [16]: [(x + y + z).match(Wild('p', exclude=i) + Wild('q', exclude=set([x, y, z]) - set(i))) for i in subsets([x, y, z])]
Out[16]: [{p: x + y + z, q: 0}, {p: y + z, q: x}, {p: x + z, q: y}, {p: x + y, q: z}, {p: z, q: x + y}, {p: y, q: x + z}, {p: x, q: y + z}, {p: 0, q: x + y + z}]

(subsets 是来自 sympy.utilities.iterables 的 SymPy 函数)。