我們這次實現(xiàn)的命令行計算器,支持加減乘除褒链、括號、浮點數(shù)疑苔、負數(shù)甫匹,以及查看歷史和退出功能。
主要的思路:read - parse - print - loop惦费。
read 階段是指從讀取用戶在提示符(cal> )之后輸入的字符串兵迅。
parse 階段包括:將用戶輸入的字符串分割成單個對象比如:符號 +,或者數(shù)字 1.2 薪贫。其次將中綴表達式轉(zhuǎn)化后綴表達式恍箭,接下來計算后綴表達式的數(shù)值。
print 階段指的是將 parse 階段的數(shù)值打印在終端上瞧省。
loop 階段是指重復上述操作扯夭,除非用戶輸入 exit 符號或者 Ctrl + C 終止程序運行鳍贾。
實現(xiàn)效果圖:
棧
為了實現(xiàn)后面的中綴、后綴表達式交洗,我們需要借助棧這種數(shù)據(jù)結(jié)構(gòu)骑科。
棧是一種先進后出的數(shù)據(jù)結(jié)構(gòu),函數(shù)調(diào)用時就會使用棧來保存當前程序的地址及其他信息构拳,前面的深度優(yōu)先搜索就是基于棧實現(xiàn)的咆爽。
之前寫了一篇文章介紹了棧的用法,下面簡單說明棧的使用:
push 函數(shù)在棧的頂部添加元素隐圾,pop 函數(shù)刪除棧頂?shù)脑匚橄疲瑃op 函數(shù)查看棧頂函數(shù)。
class Stack:
def push(self, value: T) -> None: pass
def pop(self) -> T: pass
def top(self) -> T: pass
分詞
分詞是 parse 階段的重要部分暇藏。
我們定義符號 + - * / ( )蜜笤。
將這些字符串中的這些符號替換為:空格 + 符號 + 空格
+ => 空格+空格
- 將字符串按空格分割,得到最終的分詞
經(jīng)過上述操作盐碱,就可以實現(xiàn)對正數(shù)以及操作符的分詞功能把兔。
# Python高效編程
def tokenize(expression: str) -> Generator[str, None, None]:
for symbol in symbols:
# 替換字符串
expression = expression.replace(symbol, f" {symbol} ")
# 分割字符串
為了實現(xiàn)對負數(shù)的分詞,我們需要迭代分詞序列瓮顽,將‘-’ 與后面的數(shù)字轉(zhuǎn)換成一個整體县好。我們有兩種情況將 - 視作負號,第一種情況暖混,負號位于字符串的開頭缕贡;第二種情況,負號前一個負號為左括號拣播。
# Python高效編程
for i, item in enumerate(expression.split()):
if item == "-":
# 兩種情況下晾咪,當前符號設(shè)置為 ""
# 下一個符號設(shè)置為負數(shù)
if i - 1 < 0 or seq[i - 1] == "(":
seq[i] = ""
seq[i + 1] = f"-{seq[i + 1]}"
# 若字符串不為空,打印字符串
if seq[i]:
yield seq[i]
中綴表達式
中綴表達式就是我們寫的各種表達式贮配,如 3 + 2 * 4谍倦。我們希望將其轉(zhuǎn)化為后綴表達式,示例如下:
中綴表達式 ==> 1 + 5 * 4 + (3 * 2 + 4) * 6
后綴表達式 ==> 1 5 4 * + 3 2 * 4 + 6 * +
我們需要準備一個棧 stack 泪勒,用于臨時存放操作符昼蛀;準備一個序列 seq,用來存放輸出結(jié)果圆存,也就是后綴表達式叼旋。
轉(zhuǎn)化規(guī)則:
迭代分詞序列,遇到數(shù)字沦辙,就將數(shù)字輸出到序列 seq 中夫植。
定義各個符號優(yōu)先級,其中 乘除 > 加減 > 右括號怕轿。比較當前符號與棧頂元素的優(yōu)先級偷崩,如果當前元素的優(yōu)先級小于等于棧頂元素的優(yōu)先級,就彈出棧頂元素撞羽,輸出到 seq 序列中阐斜,然后將當前元素壓入棧中。如果當前元素的優(yōu)先級大于棧頂元素的優(yōu)先級诀紊,或者當前元素為右括號谒出,我們直接將當前元素壓入棧中。
如果遇到左括號邻奠,我們就彈出棧中元素笤喳,并輸出到 seq 序列中。其中碌宴,左右括號并不輸出到序列中杀狡。
最后,將棧中剩余的符號輸出到 seq 序列中贰镣。
# Python高效編程
# 迭代 1 + 5 * 4 + (3 * 2 + 4) * 6
# 讀入 1 并輸出到 seq 中呜象,將 + 壓入棧中
# 讀入 5 并輸出到 seq 中
stack ==> +
seq ==> 1 5
# 讀入 *, * 優(yōu)先級 > + 優(yōu)先級
# 將 * 壓入棧中, 讀入 4 并輸出
stack ==> + *
seq ==> 1 5 4
# 讀入 +, + 優(yōu)先級 < * 優(yōu)先級
# 將 * 輸出到 seq 序列中
# + 優(yōu)先級 <= + 優(yōu)先級碑隆,+ 輸出
# 將 + 壓入棧中
stack ==> +
seq ==> 1 5 4 * +
# 讀入 (, 壓入棧中恭陡,讀入 3,壓入棧中
stack ==> + (
seq ==> 1 5 4 * + 3
# 讀入 *, 壓入棧中上煤,讀入 2休玩,壓入棧中
stack ==> + ( *
seq ==> 1 5 4 * + 3 2
# 讀入 +,* 輸出劫狠,+ 壓入棧中拴疤, 4 輸出
stack ==> + ( +
seq ==> 1 5 4 * + 3 2 * 4
# 讀入 ),+ 輸出嘉熊,彈出 (
stack ==> +
seq ==> 1 5 4 * + 3 2 * 4 +
# 讀入 *, 壓入棧中, 6 輸出
stack ==> + *
seq ==> 1 5 4 * + 3 2 * 4 + 6
# 將棧中剩余符號輸出
stack ==>
seq ==> 1 5 4 * + 3 2 * 4 + 6 * +
代碼如下:
# Python高效編程
def parse_infix(expression: str) -> List[str]:
stack: Stack[str] = Stack()
result: List[str] = []
for expr in tokenize(expression):
if not is_float(expr) and expr not in symbols:
raise SymbolError()
if is_float(expr):
result.append(expr)
elif expr == ")":
while stack.top != "(":
result.append(stack.pop())
stack.pop()
elif expr == "(":
stack.push(expr)
elif stack.top and compare_priority(expr, stack.top):
result.append(stack.pop())
while stack.top and compare_priority(expr, stack.top):
result.append(stack.pop())
stack.push(expr)
else:
stack.push(expr)
while stack:
result.append(stack.pop())
return result
后綴表達式
后綴表達式的規(guī)則:遇到數(shù)值就壓入棧中遥赚;遇到符號,就從棧中彈出兩個元素阐肤,計算兩個元素運算結(jié)果凫佛,將得到的結(jié)果壓入棧中。
# Python高效編程
def parse_sufix(expression: List[str]) -> float:
stack: Stack[str] = Stack()
result: float = 0.0
for expr in expression:
if is_float(expr):
stack.push(expr)
else:
elem1: str = stack.pop()
elem2: str = stack.pop()
result = evaluate(expr, float(elem2), float(elem1))
stack.push(str(result))
if stack:
result = float(stack.pop())
return result
總結(jié)
接下來還有添加查看歷史和退出功能以及整個程序的組成孕惜,關(guān)注微信公眾號python高效編程
在后臺回復 2019813
獲取源代碼及 exe 文件愧薛。