executable-engineering 0.1.42__py3-none-any.whl → 0.1.43__py3-none-any.whl

executable_engineering/__init__.py

@@ -6,7 +6,7 @@ from .openoptimization_module import OptimizerOpen
 from .gradientoptimization_module import OptimizerGrad
 from .numerical_error_module import python_internal_binary, decimal_to_binary
 from .linearsystems_module import visual_solve_2d, visualize_conditioning, visualize_matrix_norms, jacobi_step, gauss_seidel_step, sor_step, visualize_convergence_2d, create_diagonally_dominant_matrix, iter_solve, steepest_descent, conjugate_gradient, IterationTracker, discretise_poisson, spai
-from .functions_module import weierstrass, plot_and_find_roots, complex_plot_with_shadow, plot_polynomial_roots_companion
+from .functions_module import weierstrass, plot_and_find_roots, complex_plot_with_shadow, plot_polynomial_roots_companion, plot_secant, plot_newton_raphson, plot_basins_of_attraction
 from .optimization_module import visual_golden_section, visual_powell_2d, newton_method
 
 __all__ = [
@@ -14,6 +14,9 @@ __all__ = [
   "visual_powell_2d",
   "visual_golden_section",
   "plot_polynomial_roots_companion",
+  "plot_secant",
+  "plot_newton_raphson",
+  "plot_basins_of_attraction",
   "plot_and_find_roots",
   "complex_plot_with_shadow",
   "weierstrass",

executable_engineering/functions_module.py

@@ -149,3 +149,130 @@ def plot_polynomial_roots_companion(n):
     title = f"Complex Space Plot: Roots of Order {n} Polynomial"
     complex_plot_with_shadow(poly_func, title=title, roots=roots, plot_range=(-plot_limit, plot_limit))
 
+
+def plot_secant(f, x0, x1, tolerance=1e-6, max_iterations=100, title='Secant Method'):
+    import plotly.graph_objects as go
+    import plotly.express as px
+    import numpy as np
+
+    x_values = [x0, x1]
+    for i in range(max_iterations):
+        denom = f(x1) - f(x0)
+        if denom == 0:
+            break
+        x_new = x1 - f(x1) * (x1 - x0) / denom
+        x_values.append(x_new)
+        if abs(f(x_new)) < tolerance:
+            break
+        x0 = x1
+        x1 = x_new
+    
+    root = x_values[-1] if abs(f(x_values[-1])) < tolerance else None
+
+    # Determine plot bounds
+    min_x = min(x_values)
+    max_x = max(x_values)
+    padding = (max_x - min_x) * 0.2 if max_x > min_x else 0.5
+    x = np.linspace(min_x - padding, max_x + padding, 100)
+    
+    fig = go.Figure()
+    fig.add_trace(go.Scatter(x=x, y=f(x), mode='lines', name='f(x)'))
+    fig.add_trace(go.Scatter(x=[x_values[0], x_values[1]], y=[f(x_values[0]), f(x_values[1])], mode='markers', name='Initial guesses'))
+    
+    colors = px.colors.qualitative.Plotly
+    
+    for i in range(1, len(x_values) - 1):
+        # Secant line intersecting x-axis
+        fig.add_trace(go.Scatter(x=[x_values[i-1], x_values[i], x_values[i+1]], y=[f(x_values[i-1]), f(x_values[i]), 0], mode='lines', line=dict(dash='dash', color='gray'), showlegend=False))
+        # Vertical dotted line to the curve
+        fig.add_trace(go.Scatter(x=[x_values[i+1], x_values[i+1]], y=[0, f(x_values[i+1])], mode='lines', line=dict(dash='dot', color='gray'), showlegend=False))
+        
+        color = colors[(i-1) % len(colors)]
+        fig.add_trace(go.Scatter(x=[x_values[i+1]], y=[0], mode='markers', marker=dict(size=10, color=color), name=f'x_{i+1}'))
+    
+    if root:
+        fig.add_trace(go.Scatter(x=[root], y=[0], mode='markers', marker=dict(color='green', size=10), name='Approximate root'))
+    
+    fig.update_layout(title=title, xaxis_title='x', yaxis_title='f(x)')
+    fig.show()
+    return root, x_values
+
+def plot_newton_raphson(f, df, x0, tolerance=1e-6, max_iterations=100, title='Newton-Raphson Method', xrange=None, yrange=None):
+    import plotly.graph_objects as go
+    import plotly.express as px
+    import numpy as np
+    
+    x_values = [x0]
+    for i in range(max_iterations):
+        d = df(x0)
+        if d == 0:
+            break
+        x_new = x0 - f(x0) / d
+        x_values.append(x_new)
+        if abs(f(x_new)) < tolerance or abs(x_new - x0) < tolerance:
+            break
+        x0 = x_new
+
+    root = x_values[-1] if abs(f(x_values[-1])) < tolerance or abs(x_values[-1] - x_values[-2]) < tolerance else None
+
+    # Determine plot bounds
+    if xrange is None:
+        min_x = min(x_values)
+        max_x = max(x_values)
+        padding = (max_x - min_x) * 0.2 if max_x > min_x else 0.5
+        x = np.linspace(min_x - padding, max_x + padding, 100)
+    else:
+        x = np.linspace(xrange[0], xrange[1], 100)
+        
+    fig = go.Figure()
+    fig.add_trace(go.Scatter(x=x, y=f(x), mode='lines', name='f(x)'))
+    fig.add_trace(go.Scatter(x=[x_values[0]], y=[f(x_values[0])], mode='markers', name='Initial guess'))
+    
+    colors = px.colors.qualitative.Plotly
+    
+    for i in range(len(x_values) - 1):
+        # Tangent line intersecting x-axis
+        fig.add_trace(go.Scatter(x=[x_values[i], x_values[i+1]], y=[f(x_values[i]), 0], mode='lines', line=dict(dash='dash', color='gray'), showlegend=False))
+        # Vertical dotted line to the curve
+        fig.add_trace(go.Scatter(x=[x_values[i+1], x_values[i+1]], y=[0, f(x_values[i+1])], mode='lines', line=dict(dash='dot', color='gray'), showlegend=False))
+        
+        color = colors[i % len(colors)]
+        fig.add_trace(go.Scatter(x=[x_values[i+1]], y=[0], mode='markers', marker=dict(size=10, color=color), name=f'x_{i+1}'))
+    
+    if root:
+        fig.add_trace(go.Scatter(x=[root], y=[0], mode='markers', marker=dict(color='green', size=10), name='Approximate root'))
+    
+    fig.update_layout(title=title, xaxis_title='x', yaxis_title='f(x)')
+    if xrange:
+        fig.update_layout(xaxis_range=xrange)
+    if yrange:
+        fig.update_layout(yaxis_range=yrange)
+        
+    fig.show()
+    return root, x_values
+
+def plot_basins_of_attraction(f, df, known_roots, title, real_range=(-2, 2), imag_range=(-2, 2), grid_size=500, th=1e-3):
+    import numpy as np
+    import plotly.graph_objects as go
+    from scipy.optimize import newton
+
+    real_values = np.linspace(real_range[0], real_range[1], grid_size)
+    imag_values = np.linspace(imag_range[0], imag_range[1], grid_size)
+    z_grid = np.array([[complex(r, i) for r in real_values] for i in imag_values])
+
+    roots = newton(f, z_grid, fprime=df)
+
+    colors = np.zeros((grid_size, grid_size))
+    for i in range(grid_size):
+        for j in range(grid_size):
+            if roots[i, j] is None:
+                colors[i, j] = 0
+            else:
+                for idx, r in enumerate(known_roots):
+                    if abs(roots[i, j] - r) < th:
+                        colors[i, j] = idx + 1
+                        break
+                        
+    fig = go.Figure(data=go.Heatmap(z=colors, x=real_values, y=imag_values, colorscale='Viridis', showscale=False))
+    fig.update_layout(title=title, xaxis_title='Real', yaxis_title='Imaginary', width=600, height=600)
+    fig.show()

{executable_engineering-0.1.42.dist-info → executable_engineering-0.1.43.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: executable_engineering
-Version: 0.1.42
+Version: 0.1.43
 Summary: Demonstrative tools for the Executable Engineering textbook
 Home-page: https://github.com/themintlab/ExecutableEngineering/tree/main/executable_engineering
 Author: Michael Welland

{executable_engineering-0.1.42.dist-info → executable_engineering-0.1.43.dist-info}/RECORD

@@ -1,7 +1,7 @@
-executable_engineering/__init__.py,sha256=U5BflsvsuI939I5hd0cHdlo_O2ks0Wdzds9Q88SjnHE,2043
+executable_engineering/__init__.py,sha256=p_BjWnXnCdd_g2SUfgn5L9xSEBCjSLDrZV0inHka1Sc,2177
 executable_engineering/closedoptimization_module.py,sha256=i9KkdulyCZmriYKlx0CuUetjoMmxJm9ADWgStU3w85w,9273
 executable_engineering/closedrootfinder_module.py,sha256=hQ6iWNi16RkZTaE_WPUEwkohmrYrAyU8WWSXO2EPLJY,6000
-executable_engineering/functions_module.py,sha256=NsPFoVwd3h7zsyW3QFTGZxciKyLQP1vepvmwXP2GQ20,5164
+executable_engineering/functions_module.py,sha256=HmZacKzSpUtjM1BLzAoXAgnARhQXizJcWeL7zZnO5zM,10473
 executable_engineering/gradientoptimization_module.py,sha256=-2fErE9cemSEPeV9PlWOf4cF7Tr6DjilEw_NxvE-JYw,6601
 executable_engineering/linearsystems_module.py,sha256=w6UkwZu80gUz5bAIWY47xJbwaEVzR9tCcIBkMsKjIRk,18356
 executable_engineering/neuralnet_module.py,sha256=Hu9QFoMyODg4RCEpHmT9zTwzjcVFYM8ZLYtbB3CbrsA,10102
@@ -9,8 +9,8 @@ executable_engineering/numerical_error_module.py,sha256=wxfYP8u29c19I2D6MxpQV3zD
 executable_engineering/openoptimization_module.py,sha256=OCPyrynkTo1It50RoT7irnL72aQn6FexE_R1SfFYxlg,8590
 executable_engineering/openrootfinder_module.py,sha256=R_lPtCQsIGLXV-j_uDpQepImOOgtgA7I6qnhUPwWLLc,5944
 executable_engineering/optimization_module.py,sha256=qFKdksJxeU4WZ9CDKs9lF-kn-FIHUx9DhifRprlATHo,12315
-executable_engineering-0.1.42.dist-info/licenses/LICENSE,sha256=Tg4nrt8rFhgpkPbIA8SWoCR3MlGLw0NufB8AE6-TH_o,1074
-executable_engineering-0.1.42.dist-info/METADATA,sha256=nTJKqizo9OK2jPscuD6X3V4nErqZf7cq8ez31H_pkoM,1648
-executable_engineering-0.1.42.dist-info/WHEEL,sha256=YVMoNqKzERt-wjUZwJ33xBGAwnFl-4cqbYkTtWa4itE,91
-executable_engineering-0.1.42.dist-info/top_level.txt,sha256=0U5ATV5WGTLvN9rRSjQ54TyL-ztTyXaN3FzZ_QMq4HM,23
-executable_engineering-0.1.42.dist-info/RECORD,,
+executable_engineering-0.1.43.dist-info/licenses/LICENSE,sha256=Tg4nrt8rFhgpkPbIA8SWoCR3MlGLw0NufB8AE6-TH_o,1074
+executable_engineering-0.1.43.dist-info/METADATA,sha256=OEali3wRbahuFRUapelVMCs-TLz2p36HbsV_5QNbBYg,1648
+executable_engineering-0.1.43.dist-info/WHEEL,sha256=YVMoNqKzERt-wjUZwJ33xBGAwnFl-4cqbYkTtWa4itE,91
+executable_engineering-0.1.43.dist-info/top_level.txt,sha256=0U5ATV5WGTLvN9rRSjQ54TyL-ztTyXaN3FzZ_QMq4HM,23
+executable_engineering-0.1.43.dist-info/RECORD,,