lafrite/Mapytex
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

improve printing ofr FormalExp

Browse Source
This commit is contained in:
lafrite 2014-01-28 21:04:27 +01:00 committed by Lafrite
parent eece262566
commit cb198eb5f0
1 changed files with 18 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
pymath/formal.py
View File

@ -134,13 +134,27 @@ class FormalExp(object):
return len(list(self._coef.keys())) return len(list(self._coef.keys()))
def __str__(self): def __str__(self):
return " + ".join([str(v) + str(k) for k,v in self._coef.items()]) ans = ""
for k,v in self._coef.items():
if v < 0:
ans += "-"
else:
ans += "+"
if abs(v) == 1:
ans += str(k)
else:
ans += str(abs(v)) + str(k)
if ans[0] == "+":
return ans[1:]
else:
return ans
if __name__ == '__main__': if __name__ == '__main__':
fe1 = FormalExp({"x": 1, "":2}) fe1 = FormalExp({"x": -1, "":-2})
print(fe1.get_postfix()) print(fe1)
fe2 = FormalExp({"x^12": 5, "":2}) fe2 = FormalExp({"x^12": 5, "":2})
print(fe2.get_postfix()) print(fe2)
fe3 = fe1 * fe2 fe3 = fe1 * fe2
for s in fe3: for s in fe3:
print(s) print(s)