python多分支实现四则运算器_python 四则运算计算器

刚开始只是为了实现某个功能，中间需要对表达式进行计算，想着写个四则运算计算器也很简单就顺手写了， 写到后面发现需要考虑的问题还挺多的，囧。

现在把代码贴出来记录一下这辛苦的过程吧，毕竟写得挺长的= =

不做解释了，单纯为了记录，毕竟写的也挺菜的，也就能基本实现功能2333

def calculate(in_formula: str) -> int:

"""

对字符串类型的数学表达式进行计算，返回计算结果

数学表达式可包含加减乘除、括号

:param in_formula: 数学表达式 如'2*(-5+3)/-1+20/(3+1)+2+(-9-2)*2'

:return: 计算结果

"""

if '(' in in_formula:

result_in_brackets = []

reg_formula_in_brackets = re.compile(r'\([^(]+\)')

formula_in_brackets = reg_formula_in_brackets.findall(in_formula)

else_formula = reg_formula_in_brackets.split(in_formula)

for i in formula_in_brackets:

result_in_brackets.append(simple_arithmetic(re.sub(r"[()]", '', i)))

for i in range(0, int(len(else_formula)) + 1, 2):

else_formula.insert(i + 1, str(result_in_brackets[int(i / 2)]))

result = simple_arithmetic(''.join(else_formula))

else:

result = simple_arithmetic(''.join(in_formula))

return result

def simple_arithmetic(in_formula: str) -> int:

"""

不含括号的简单公式计算（加减乘除）

:param in_formula: 只包含加减乘除不含括号的计算公式

:return: 计算结果

"""

result = 0

if '+' in in_formula or '-' in in_formula:

reg = re.compile(r"(?

operators = reg.findall(in_formula)

sub_formulas = reg.split(in_formula)

for ind, sub_formula in enumerate(sub_formulas):

if ind == 0 and sub_formula == '':

sub_formulas[0] = 0

sub_formulas[1] = operators[0] + sub_formulas[1]

operators[0] = '+'

continue

if '/' in sub_formula or '*' in sub_formula:

val = multiply_divide(sub_formula)

else:

val = int(sub_formula)

sub_formulas[ind] = val

result = sub_formulas[0]

for ind, i in enumerate(operators):

if i == '+':

result += int(sub_formulas[ind + 1])

elif i == '-':

result -= int(sub_formulas[ind + 1])

elif '/' in in_formula or '*' in in_formula:

result = multiply_divide(in_formula)

else:

return result

return result

def multiply_divide(in_formula: str) -> int:

"""

乘除法的计算

:param in_formula:只含有乘除法的计算公式

:return: 计算结果

"""

result = 1

if '/' not in in_formula:

sub_formulas = in_formula.split('*')

for i in sub_formulas:

result = result * int(i)

elif '*' not in in_formula:

sub_formulas = in_formula.split('/')

for ind, i in enumerate(sub_formulas):

result = int(i) if ind == 0 else result / int(i)

else:

reg = re.compile(r'[*/]')

operators = reg.findall(in_formula)

sub_formulas = reg.split(in_formula)

result = int(sub_formulas[0])

for ind, i in enumerate(operators):

if i == '*':

result = int(sub_formulas[ind + 1]) * result

elif i == '/':

result = result / int(sub_formulas[ind + 1])

return result

def addition_subtraction(formula: str) -> int:

"""

加减法运算

:param formula:只含有加减法的运算公式

:return: 运算结果

"""

result = 0

if '+' not in formula:

sub_formulas = formula.split('-')

result = int(sub_formulas[0])

for i in sub_formulas[1:]:

result = result - int(i)

elif '-' not in formula:

sub_formulas = formula.split('+')

for i in sub_formulas:

result = int(i) + result

elif '+' in formula or '-' in formula:

reg = re.compile(r'[+\-]')

operators = reg.findall(formula)

sub_formulas = reg.split(formula)

result = int(sub_formulas[0])

for ind, i in enumerate(operators):

if i == '+':

result += int(sub_formulas[ind + 1])

elif i == '-':

result -= int(sub_formulas[ind + 1])

else:

return formula

return result

if __name__ == '__main__':

print(calculate('2*(-5+3)/-1+20/(3+1)+2+(-9-2)*2'))