Merge pull request AllenDowney#16 from Bogidon/bogdan/fsolve-array

AllenDowney · web-flow · commit 72b855771dde · 2017-11-29T07:06:18.000-05:00
Update fsolve to work with arrays
diff --git a/code/modsim.py b/code/modsim.py
@@ -347,6 +347,8 @@ def fsolve(func, x0, *args, **kwargs):
     returns: solution as an array
     """
     # make sure we can run the given function with x0
+    x0 = np.asarray(x0).flatten() 
+    
     try:
         func(x0, *args)
     except Exception as e:
