Tiny Tiny Edit to Thomas Algorithm in Python (#377)

* Adding A Single Character

* Adding other spaces

* Adding Spaces Between '#' and the Comment

* Adding Blank Comment Line

* Removing Line Ending Whitespace

* Minor Code Cleanup

* Changing Comment to Blank Line

Author: Wesley-Arrington

contents/thomas_algorithm/code/python/thomas.py

@@ -2,31 +2,32 @@
 
 # note this example is inplace and destructive
 def thomas(a, b, c, d):
-    
-    #set the initial elements
+
+    # set the initial elements
     c[0] = c[0] / b[0]
     d[0] = d[0] / b[0]
 
-    n = len(d) #number of equations to solve
-    for i in range(1,n):
-        #scale factor for c and d
-        scale = 1 / (b[i] - c[i-1]*a[i])
-        
-        c[i] = c[i] * scale
-        d[i] = (d[i] -a[i] * d[i-1]) * scale
+    n = len(d) # number of equations to solve
+    for i in range(1, n):
+        # scale factor for c and d
+        scale = 1 / (b[i] - c[i-1] * a[i])
+
+        c[i] *= scale
+        d[i] = (d[i] - a[i] * d[i-1]) * scale
+
 
-    
     # do the back substitution
-    for i in range(n-2,-1,-1):
-        d[i] = d[i] - c[i]*d[i+1]
-    
+    for i in range(n-2, -1, -1):
+        d[i] -= c[i] * d[i+1]
+
     return d
 
 def main():
     # example for matrix
     # [1  4  0][x]   [7]
     # [2  3  5][y] = [5]
     # [0  3  6][z]   [3]
+
     #                 [.8666]
     # soln will equal [1.533]
     #                 [-.266]
@@ -35,6 +36,7 @@ def main():
     b = [1, 3, 6]
     c = [4, 5, 0]
     d = [7, 5, 3]
+
     soln = thomas(a, b, c, d)
     print(soln)