如何重新排列包含关系运算符的sympy表达式

时间:2015-05-13 21:17:23

标签: python sympy

我有包含关系运算符,符号和常量的表达式。我想重新排列表达式,以便(在可能的范围内)所有常量项都在关系运算符的一侧,而剩余的项在另一侧。例如,我想重新排列:

x - 5> y - z

为:

x - y + z> 5

是否存在这样做的同意方法?如果没有,我应该从哪里开始延伸同情?

2 个答案:

答案 0 :(得分:1)

有些令人惊讶的是,我无法找到一种“开箱即用”的方法。您可以使用this question中的方法使任何一个变量成为不等式左侧(LHS)的唯一主题,但我似乎无法使常数项成为主语。

所以,我写了自己的版本并在下面重现它。我已经根据给出的示例和其他一些示例对其进行了测试。它试图使右侧(RHS)由零或仅基于可选参数的常数项组成。可能存在失败的极端情况 - 谨慎使用/修改。

代码:

import sympy
from sympy.core.relational import Relational

mult_by_minus_one_map = {
    None: '==',
    '==': '==',
    'eq': '==',
    '!=': '!=',
    '<>': '!=',
    'ne': '!=',
    '>=': '<=',
    'ge': '<=',
    '<=': '>=',
    'le': '>=',
    '>': '<',
    'gt': '<',
    '<': '>',
    'lt': '>',
}

def move_inequality_constants(ineq, zero_on_right=False):
    l = ineq.lhs
    r = ineq.rhs
    op = ineq.rel_op
    all_on_left = l - r
    if zero_on_right:
        return Relational(all_on_left, sympy.sympify(0), op)
    else:
        coeff_dict = all_on_left.as_coefficients_dict()
        var_types = coeff_dict.keys()
        new_rhs = sympy.sympify(0)
        for s in var_types:
            if s == 1:
                all_on_left = all_on_left - coeff_dict[s]
                new_rhs = new_rhs - coeff_dict[s]
        if new_rhs < 0:
            all_on_left = all_on_left * -1
            new_rhs = new_rhs * -1
            op = mult_by_minus_one_map[op]
        return Relational(all_on_left,new_rhs,op)

# test code to demo function below    
from sympy.abc import x,y,z

test_ineqs = [ x - 5 > y - z,
               x**2 + x - 5 > y + x**2 - z,
               x + 5 > y - z,
               x**3 + y**2 >= x + 5*y - z - 15]

for k in test_ineqs:
    print('Re-arranging '+ str(k))
    kn = move_inequality_constants(k)
    print('Gives '+str(kn))
    print('Or equivalently ' + str(move_inequality_constants(k, True)))
    print('====')

输出:

Re-arranging x - 5 > y - z
Gives x - y + z > 5
Or equivalently x - y + z - 5 > 0
====
Re-arranging x**2 + x - 5 > x**2 + y - z
Gives x - y + z > 5
Or equivalently x - y + z - 5 > 0
====
Re-arranging x + 5 > y - z
Gives -x + y - z < 5
Or equivalently x - y + z + 5 > 0
====
Re-arranging x**3 + y**2 >= x + 5*y - z - 15
Gives -x**3 + x - y**2 + 5*y - z <= 15
Or equivalently x**3 - x + y**2 - 5*y + z + 15 >= 0

答案 1 :(得分:0)

canonical方法中省略这一点可能是一种疏忽。

以下内容可能适用于您的else子句

r, l = (ineq.lhs - ineq.rhs).as_coeff_Add()
if r < 0:
    l, r = -r, -l
return Relational(l, r, ineq.rel_op).canonical

有人可能会想到规范会删除常见因素,因此2x<4y会变为x<2*y。 SymPy将对实现此类更改的pull请求持开放态度。