50行代码实现python计算器主要功能

时间:2022-04-16 22:53:20

实现功能:计算带有括号和四则运算的式子   3*( 4+ 50 )-(( 100 + 40 )*5/2- 3*2* 2/4+9)*((( 3 + 4)-4)-4)

基本思路:使用正则表达式提取出每一层小括号式子,计算出结果后替换该括号,直至最后一层。

0、计算前的准备,进行式子的整理,如果非法则退出。这里我只对空格和符号进行了处理,真实场景中应该做更多的判断:

def check(s):
s = s.replace(" ", '')
s = re.sub(r'(\+-)|(-\+)', '-', s)
s = re.sub(r'(\+\+)|(--)', '+', s)
return s

1、只有加减运算的式子:findall匹配后直接使用sum函数对list求和,此时的减号看作负号,加号看作了正号,所以代码极其简洁:

import re

def plus_all(s): #实现加减运算
patter_plus = r'[+-]?\d+\.?\d*'
ret = re.findall(patter_plus, s)
return sum([float(x) for x in ret])

2、只有乘除运算的式子:和以上思路一致,主要想把除法(/)转换为乘法后用reduce计算。其先匹配出带有乘除运算符的list,然后通过change_mul_div()函数将除法转换成倒数计算

from functools import reduce

def mul_all(s):
patter_mul = r'[*/]?[-]?\d+\.?\d*'
ret = re.findall(patter_mul, s)
return reduce(lambda m, n: m*n, [change_mul_div(x) for x in ret]) def change_mul_div(s):
if "/" in s:
return 1 / float(s.replace('/', ""))
else:
return float(s.replace("*", ''))

3、含有加减乘除但不含有括号的式子:先用mul_all(s)函数将式子中的所有乘除法计算完毕,然后回到第1步。cal_all_mul_div函数实现计算式子中所有乘除法:

def cal_all_mul_div(s):
patter_all_mul_div = r'([+-]?\d+\.?\d*[*/])+[+-]?\d+\.?\d*'
ret = re.search(patter_all_mul_div, s)
while ret:
mul_div = mul_all(ret.group())
s = s.replace(ret.group(), ''.join(['+', str(mul_div)]))
ret = re.search(patter_all_mul_div, s)
return s

4、不断提取括号进行计算:

def bracket(s):
patter_kuohao = r'\([^()]+\)'
ret = re.search(patter_kuohao, s)
while ret:
cal_result = plus_all(cal_all_mul_div(ret.group()))
s=s.replace(ret.group(), ''.join(['+', str(cal_result)]))
ret = re.search(patter_kuohao, s)
return s

所有代码合并如下:

import re
from functools import reduce s3 = "3*( 4+ 50 )-(( 100 + 40 )*5/2- 3*2* 2/4+9)*((( 3 + 4)-4)-4)" def check(s):
s = s.replace(" ", '') #对空格进行处理
s = re.sub(r'(\+-)|(-\+)', '-', s) #对符号进行处理
s = re.sub(r'(\+\+)|(--)', '+', s)
return s def plus_all(s):
patter_plus = r'[+-]?\d+\.?\d*'
ret = re.findall(patter_plus, s) #找出所有的加数
return sum([float(x) for x in ret]) #生成器生成列表并求和 def mul_all(s):
patter_mul = r'[*/]?[-+]?\d+\.?\d*'
ret = re.findall(patter_mul, s)
return reduce(lambda m, n: m * n, [change_mul_div(x) for x in ret]) #reduce连乘 def change_mul_div(s): #把除法改成乘以其倒数
if "/" in s:
return 1 / float(s.replace('/', ""))
else:
return float(s.replace("*", '')) def cal_all_mul_div(s):
patter_all_mul_div = r'([+-]?\d+\.?\d*[*/])+[+-]?\d+\.?\d*'
ret = re.search(patter_all_mul_div, s)
while ret:
mul_div = mul_all(ret.group())
s = s.replace(ret.group(), ''.join(['+', str(mul_div)]))
ret = re.search(patter_all_mul_div, s)
return s def bracket(s):
patter_kuohao = r'\([^()]+\)'
s = check(s)
ret = re.search(patter_kuohao, s)
while ret:
cal_result = plus_all(cal_all_mul_div(ret.group()))
s = check(s.replace(ret.group(), ''.join(['+', str(cal_result)]))) #替换时需要在前面加入正号,并对s进行check操作
ret = re.search(patter_kuohao, s)
return s s3 = ''.join(["(", s3, ")"])
print(float(bracket(s3)))

以上即用不超过50行代码实现了计算器的主要功能。