no more string in expression manipulation only list of tokens. That's

great!
This commit is contained in:
Lafrite 2013-10-19 20:33:46 +02:00
parent 0423412e6b
commit b2d8991107
1 changed files with 77 additions and 61 deletions

View File

@ -4,31 +4,41 @@
from generic import Stack from generic import Stack
def str2tokens(exp):
"""Convert an expression into a list of tokens
:param exp: The expression
:returns: the list of tokens
def infixToPostfix(infixExp): >>> str2tokens("1 + 2")
"""Transform an infix expression into postfix expression ['1', '+', '2']
:param infixExp: an infix expression (caracters must be separate by a space) """
:returns: the corresponding postfix expression return exp.split(" ")
def infixToPostfix(infixTokens):
"""Transform an infix list of tokens into postfix tokens
:param infixTokens: an infix list of tokens
:returns: the corresponding postfix list of tokens
:Example: :Example:
>>> infixToPostfix("1 + 2") >>> infixToPostfix([1, "+", 2])
'1 2 +' [1, 2, '+']
>>> infixToPostfix("1 * 2 + 3")
'1 2 * 3 +'
>>> infixToPostfix([1, "*", 2, "+", 3])
[1, 2, '*', 3, '+']
""" """
priority = {"*" : 3, "/": 3, "+": 2, "-":2, "(": 1} priority = {"*" : 3, "/": 3, "+": 2, "-":2, "(": 1}
opStack = Stack() opStack = Stack()
postfixList = [] postfixList = []
tokenList = infixExp.split(" ") #infixTokens = infixExp.split(" ")
for token in tokenList: for token in infixTokens:
if token == "(": if token == "(":
opStack.push(token) opStack.push(token)
elif token == ")": elif token == ")":
@ -36,7 +46,8 @@ def infixToPostfix(infixExp):
while topToken != "(": while topToken != "(":
postfixList.append(topToken) postfixList.append(topToken)
topToken = opStack.pop() topToken = opStack.pop()
elif token in "+-*/": elif type(token) == str and token in "+-*/":
# On doit ajouter la condition == str sinon python ne veut pas tester l'appartenance à la chaine de caractère.
while (not opStack.isEmpty()) and (priority[opStack.peek()] >= priority[token]): while (not opStack.isEmpty()) and (priority[opStack.peek()] >= priority[token]):
postfixList.append(opStack.pop()) postfixList.append(opStack.pop())
opStack.push(token) opStack.push(token)
@ -46,22 +57,22 @@ def infixToPostfix(infixExp):
while not opStack.isEmpty(): while not opStack.isEmpty():
postfixList.append(opStack.pop()) postfixList.append(opStack.pop())
return " ".join(postfixList) return postfixList
def computePostfix(postfixExp): def computePostfix(postfixTokens):
"""Compute a postfix expression """Compute a postfix list of tokens
:param postfixExp: a postfix expression :param postfixTokens: a postfix list of tokens
:returns: the result of the expression :returns: the result of the calculus
""" """
print(postfixToInfix(postfixExp)) #print(postfixToInfix(postfixExp))
# where to save numbers or # where to save numbers or
operandeStack = Stack() operandeStack = Stack()
tokenList = postfixExp.split(" ") #tokenList = postfixExp.split(" ")
for (i,token) in enumerate(tokenList): for (i,token) in enumerate(postfixTokens):
if token in "+-*/": if token in "+-*/":
op2 = operandeStack.pop() op2 = operandeStack.pop()
op1 = operandeStack.pop() op1 = operandeStack.pop()
@ -70,8 +81,8 @@ def computePostfix(postfixExp):
#print("Operation: {op1} {op} {op2}".format(op1 = op1, op = token, op2 = op2)) #print("Operation: {op1} {op} {op2}".format(op1 = op1, op = token, op2 = op2))
#print(operandeStack) #print(operandeStack)
#print(tokenList[i+1:]) #print(postfixTokens[i+1:])
newPostfix = " ".join(operandeStack + tokenList[i+1:]) newPostfix = " ".join(operandeStack + postfixTokens[i+1:])
print(postfixToInfix(newPostfix)) print(postfixToInfix(newPostfix))
else: else:
@ -79,18 +90,19 @@ def computePostfix(postfixExp):
return operandeStack.pop() return operandeStack.pop()
def computePostfixBis(postfixExp): def computePostfixBis(postfixTokens):
"""Compute a postfix expression like a good student """Compute a postfix list of tokens like a good student
:param postfixExp: a postfix expression :param postfixTokens: a postfix list of tokens
:returns: the result of the expression :returns: the result of the expression
""" """
print(postfixToInfix(postfixExp)) #print(postfixToInfix(postfixTokens))
# where to save numbers or # where to save numbers or
operandeStack = Stack() operandeStack = Stack()
tokenList = postfixExp.split(" ") #tokenList = postfixExp.split(" ")
tokenList = postfixTokens.copy()
# On fait le calcul jusqu'à n'avoir plus qu'un élément # On fait le calcul jusqu'à n'avoir plus qu'un élément
while len(tokenList) > 1: while len(tokenList) > 1:
@ -115,7 +127,7 @@ def computePostfixBis(postfixExp):
tmpTokenList += tokenList tmpTokenList += tokenList
tokenList = expand_list(tmpTokenList) tokenList = expand_list(tmpTokenList)
print(postfixToInfix(" ".join(tokenList))) print(postfixToInfix(tokenList))
return tokenList[0] return tokenList[0]
@ -145,18 +157,20 @@ def doMath(op, op1, op2):
return str(eval(op1 + op + op2)) return str(eval(op1 + op + op2))
def postfixToInfix(postfixExp): def postfixToInfix(postfixTokens):
"""Transforme postfix expression into infix expression """Transforme postfix list of tokens into infix list of tokens
:param postfixExp: a postfix expression :param postfixTokens: a postfix list of tokens
:returns: the corresponding infix expression :returns: the corresponding infix list of tokens
""" """
operandeStack = Stack() operandeStack = Stack()
tokenList = postfixExp.split(" ") #print(postfixTokens)
for (i,token) in enumerate(tokenList): #tokenList = postfixExp.split(" ")
for (i,token) in enumerate(postfixTokens):
if token in "+-*/": if token in "+-*/":
op2 = operandeStack.pop() op2 = operandeStack.pop()
if needPar(op2, token, "after"): if needPar(op2, token, "after"):
@ -181,6 +195,7 @@ def needPar(operande, operator, posi = "after"):
:param posi: "after"(default) if the operande will be after the operator, "before" othewise :param posi: "after"(default) if the operande will be after the operator, "before" othewise
:returns: bollean :returns: bollean
""" """
priority = {"*" : 3, "/": 3, "+": 2, "-":2} priority = {"*" : 3, "/": 3, "+": 2, "-":2}
if isNumber(operande) and "-" in operande: if isNumber(operande) and "-" in operande:
@ -197,10 +212,10 @@ def needPar(operande, operator, posi = "after"):
else: else:
return 0 return 0
def get_main_op(exp): def get_main_op(tokens):
"""Getting the main operation of the expression """Getting the main operation of the list of tokens
:param exp: the expression :param exp: the list of tokens
:returns: the main operation (+, -, * or /) or 0 if the expression is only one element :returns: the main operation (+, -, * or /) or 0 if the expression is only one element
""" """
@ -208,15 +223,15 @@ def get_main_op(exp):
priority = {"*" : 3, "/": 3, "+": 2, "-":2} priority = {"*" : 3, "/": 3, "+": 2, "-":2}
parStack = Stack() parStack = Stack()
tokenList = exp.split(" ") #tokenList = exp.split(" ")
if len(tokenList) == 1: if len(tokens) == 1:
# Si l'expression n'est qu'un élément # Si l'expression n'est qu'un élément
return 0 return 0
main_op = [] main_op = []
for token in tokenList: for token in tokens:
if token == "(": if token == "(":
parStack.push(token) parStack.push(token)
elif token == ")": elif token == ")":
@ -273,7 +288,8 @@ def test(exp):
""" """
print("-------------") print("-------------")
print("Expression ",exp) print("Expression ",exp)
postfix = infixToPostfix(exp) tokens = str2tokens(exp)
postfix = infixToPostfix(tokens)
#print("Postfix " , postfix) #print("Postfix " , postfix)
#print(computePostfix(postfix)) #print(computePostfix(postfix))
#print("Bis") #print("Bis")
@ -283,27 +299,27 @@ def test(exp):
if __name__ == '__main__': if __name__ == '__main__':
#exp = "1 + 3 * 5" exp = "1 + 3 * 5"
#test(exp) test(exp)
#exp = "2 * 3 * 3 * 5" exp = "2 * 3 * 3 * 5"
#test(exp) test(exp)
#exp = "2 * 3 + 3 * 5" exp = "2 * 3 + 3 * 5"
#test(exp) test(exp)
exp = "2 * ( 3 + 4 ) + 3 * 5"
test(exp)
exp = "2 * ( 3 + 4 ) + ( 3 - 4 ) * 5"
test(exp)
exp = "2 * ( 2 - ( 3 + 4 ) ) + ( 3 - 4 ) * 5"
test(exp)
exp = "2 * ( 2 - ( 3 + 4 ) ) + 5 * ( 3 - 4 )"
test(exp)
#exp = "2 * ( 3 + 4 ) + 3 * 5"
#test(exp)
#
#exp = "2 * ( 3 + 4 ) + ( 3 - 4 ) * 5"
#test(exp)
#
#exp = "2 * ( 2 - ( 3 + 4 ) ) + ( 3 - 4 ) * 5"
#test(exp)
#
#exp = "2 * ( 2 - ( 3 + 4 ) ) + 5 * ( 3 - 4 )"
#test(exp)
#
exp = "2 + 5 * ( 3 - 4 )" exp = "2 + 5 * ( 3 - 4 )"
test(exp) test(exp)
@ -321,8 +337,8 @@ if __name__ == '__main__':
## Ce denier pose un soucis. Pour le faire marcher il faudrai implémenter le calcul avec les fractions ## Ce denier pose un soucis. Pour le faire marcher il faudrai implémenter le calcul avec les fractions
#exp = "( 2 + 5 ) / 3 * 4" #exp = "( 2 + 5 ) / 3 * 4"
#test(exp) #test(exp)
#import doctest import doctest
#doctest.testmod() doctest.testmod()
# ----------------------------- # -----------------------------