将字符串表达式转换为数学表达

时间:2016-09-30 23:13:01

标签: python string python-3.x

我正在寻找将数学表达式字符串作为输入的东西,并通过添加括号来改变它。

例如:

输入1:"2*4^7"

输出1:"(2*(4^7))"

输入2:"sin(x)*3-8^9"

输出2:"((sin(x)*3)-(8^9))"

输入3:"(2-8)/6"

输出3:"((2-8)/6)"

我目前正在尝试对其进行编码,但它花费了太多时间而我宁愿专注于其他事情,所以如果你们知道一些可以做到这一点的模块,那就太棒了。

我正在使用python 3

谢谢!

1 个答案:

答案 0 :(得分:2)

使用ast模块,您可以解析字符串,然后遍历它以构建所需的字符串表示形式:

import ast

def represent(tree):
    t = type(tree)
    if t == ast.Module:
        return represent(tree.body)
    elif t == list:
        # arithmetic expression should only be one thing
        return ", ".join(represent(part) for part in tree)
    elif t == ast.Expr:
        return represent(tree.value)
    elif t == ast.BinOp:
        o = type(tree.op)
        if o == ast.BitXor:
            op = "^"
        elif o == ast.Mult:
            op = "*"
        elif o == ast.Add:
            op = "+"
        ...
        return "({}{}{})".format(
                represent(tree.left),
                op,
                represent(tree.right),
                )
    elif t == ast.Num:
        return str(tree.n)
    elif t == ast.Name:
        return tree.id
    elif t == ast.Call:
        return "{}({})".format(
                represent(tree.func),
                represent(tree.args),
                )
    ...

# prints (((sin(x)*2)*x)+4)
print(represent(ast.parse("sin(x)*2*x+4")))

正如@ user2357112在评论中指出的那样,只有在限制自己使用Python的表达式语法时才会这样做,最重要的是使用**代替^进行取幂。