本文介绍怎样用正则表达式实现一个简单的计算器。(文章中的代码为Python3)
1.问题描述
用正则表达式实现一个简单计算器的功能:输入为一个字符串(含有加减乘除,正负和整数及浮点数的计算公式),输出为运算结果。
2.想要达到的效果
输入计算公式:1.2+-(2*4+(2+5))-3.4,结果为:-10.4。也就是说我们允许用户输入整数和小数,计算包括加减乘除,当然如果用户想要注明数的正负也是可以识别的。
3.解决思路
3.1明确优先级
(1)():小括号的优先级最高。括号的嵌套结构中,内部的括号的优先级高于外部的括号。
(2)*或/:乘除号的优先级低于小括号,但是高于加减。
(3)+或-:加减号的优先级最低。
3.2可能包含的函数
(1)用户输入函数:这个函数在程序启动的时运行:包括提示用户输入;获得用户输入;对用户的输入进行简单的合法性验证;将用户的输入转换成标准的模式。
(2)去括号:这个是按照优先级的顺序,首先去除表达式中的括号(括号中的字符用计算结果代替)。这个函数需要调用运算函数。
(3)去乘除:这个函数是按照优先级的顺序得到乘除运算的结果,并将结果带入计算的公式中。
(4)去加减:这个函数用于加减计算。
经过上面的各个函数,用户输入的计算公式将会得到唯一的运算结果。
具体代码如下所示(python3):
__author__ = "Allen Liu"
__time__ = "2017/7/29"
'''This program is used to calculate the result of the formula using re '''
import re # 加载re模块
import sys # 加载sys模块用于结束程序
# 定义用户输入函数
def user_input():
''' 用户输入函数,包括提示信息并得到用户输入 '''
print("Welcome use this calculator".center(50,'-'))# 输出欢迎语句
input_formula = input("Please input your formula or input q to exit:")# 提示用户输入
if input_formula.lower() == 'q':# 判断用户输入是否为q(退出程序)
sys.exit('Looking forward to your re-use!')
elif input_legal(input_formula) == False:# 简单的判断下输入是否合法
sys.exit('Sorry, your input is illegal!')
else:
normal_formula = input_normal(input_formula)
print(normal_formula)
return normal_formula
# 判断用户的输入是否合法,合法的时候返回True,否则返回False
def input_legal(input_formula):
''' 判断用户输入是否有其他非法字符,有则返回False,否则为True '''
indicator = re.search(r'[^(0123456789\.\+\-\*\/\(\)\s)]',input_formula)
if indicator == None:
indicator = True
else:
indicator = False
return indicator
# 规范化用户的输入,去掉空格和不必要的正负号
def input_normal(input_formula):
''' 将用户输入变成比较规范的格式 '''
iput = re.sub('\s*', '', input_formula) # 去除空格
input_formula = input_formula.replace('+-', '-') # 替换表达式里的所有'+-'
input_formula = input_formula.replace('--', '+') # 替换表达式里的所有'--'
input_formula = input_formula.replace('-+', '-') # 替换表达式里的所有'-+'
input_formula = input_formula.replace('++', '+') # 替换表达式里的所有'++'
return input_formula
# 去括号函数
def del_bra(expression):
''' 递归的去除括号,输入为含有括号的表达式,输出为去括号后的表达式 '''
if not re.search(r'\(([^()]+)\)', expression): # 判断小括号,如果不存在小括号,直接调用乘除,加减计算
ret1 = mul_div(expression)
ret2 = add_min(ret1)
return ret2 # 返回最终计算结果
data = re.search(r'\(([^()]+)\)', expression).group() # 如果有小括号,匹配出优先级最高的小括号
# print("获取表达式",data)
data = data.strip('[\(\)]') # 剔除小括号
ret1 = mul_div(data) # 计算乘除
# print("全部乘除计算完后的表达式:",ret1)
ret2 = add_min(ret1) # 计算加减
# print("全部加减计算结果:",ret2)
part1, replace_str, part2 = re.split(r'\(([^()]+)\)', expression, 1) # 将小括号计算结果替换回表达式
expression = '%s%s%s' % (part1, ret2, part2) # 生成新的表达式
return del_bra(expression) # 递归去小括号
# 定义乘除运算函数
def mul_div(expression):
''' 去除乘除运算,返回没有乘除运算的表达式。输入为去括号后的只含有加减乘除运算的表达式,输出为只含有加减运算的表达式 '''
val = re.search('\d+\.*\d*[\*\/]+[\+\-]?\d+\.*\d*', expression) # 匹配乘除号
if not val: # 乘除号不存在,返回输入的表达式
return expression
data = re.search('\d+\.*\d*[\*\/]+[\+\-]?\d+\.*\d*', expression).group() # 匹配乘除号
if len(data.split('*')) > 1: # 当可以用乘号分割,证明有乘法运算
part1, part2 = data.split('*') # 以乘号作为分割符
value = float(part1) * float(part2) # 计算乘法
else:
part1, part2 = data.split('/') # 用除号分割
if float(part2) == 0: # 如果分母为0,则退出计算
sys.exit("Sorry,the divisor equals zero!")
value = float(part1) / float(part2) # 计算除法
# 获取第一个匹配到的乘除计算结果value,将value放回原表达式
s1, s2 = re.split('\d+\.*\d*[\*\/]+[\+\-]?\d+\.*\d*', expression, 1) # 分割表达式
# print("上一个表达式:",expression)
next_expression = "%s%s%s" % (s1, value, s2) # 将计算结果替换会表达式
# print("下一个表达式%s" % next_expression)
return mul_div(next_expression) # 递归表达式
# 定义加减函数,用于表达式中的加减计算
def add_min(expression):
'''计算公式中的加减运算,输入为只含有加减运算的公式,输出为运算结果'''
# 格式化表达式,因为这时候可能还还有正负号
expression = expression.replace('+-', '-') # 替换表达式里的所有'+-'
expression = expression.replace('--', '+') # 替换表达式里的所有'--'
expression = expression.replace('-+', '-') # 替换表达式里的所有'-+'
expression = expression.replace('++', '+') # 替换表达式里的所有'++'
data = re.search('\d+\.*\d*[\+\-]{1}\d+\.*\d*', expression) # 匹配加减号
if not data: # 如果不存在加减号,则证明表达式已计算完成,返回最终结果
return expression
val = re.search('[\-]?\d+\.*\d*[\+\-]{1}\d+\.*\d*', expression).group()
if len(val.split('+')) > 1: # 以加号分割成功,有加法计算
part1, part2 = val.split('+')
value = float(part1) + float(part2) # 计算加法
elif val.startswith('-'): # 如果是已'-'开头则需要单独计算
part1, part2, part3 = val.split('-')
value = -float(part2) - float(part3) # 计算以负数开头的减法
else:
part1, part2 = val.split('-')
value = float(part1) - float(part2) # 计算减法
s1, s2 = re.split('[\-]?\d+\.*\d*[\+\-]{1}\d+\.*\d*', expression, 1) # 分割表达式
# print("计算%s=%s" % (val,value))
next_expression = "%s%s%s" % (s1, value, s2) # 将计算后的结果替换回表达式,生成下一个表达式
# print("下一个表达式: ",next_expression)
return add_min(next_expression) # 递归运算表达式
# 主函数
if __name__ == "__main__":
'''将计算结果与eval()函数的结果进行比较,来确定计算的正确性'''
try:
expression = user_input() # 获取用户输入的表达式
reslut = eval(expression) # 用eval计算验证
ret = del_bra(expression) # 用函数计算后得出的结果
reslut = float(reslut)
ret = float(ret)
if reslut == ret: # 将两种方式计算的结果进行比较,如果相等,则计算正确,输出结果
print("eval计算结果:%s" % reslut)
print("表达式计算结果:%s" % ret)
else: # 两种计算方式的结果不正确,提示异常,并返回两种方式的计算结果
print("计算结果异常,请重新检查!")
print("eval计算结果:%s" % reslut)
print("表达式计算结果:%s" % ret)
except(SyntaxError, ValueError, TypeError): # 如果有不合法输出,则抛出错误
print("输入表达式不合法,请重新检查!")