Use np.interp instead of interpolation.interp

Author: kp992

lectures/cake_eating_numerical.md

@@ -15,15 +15,6 @@ kernelspec:
 :depth: 2
 ```
 
-In addition to what's in Anaconda, this lecture will require the following library:
-
-```{code-cell} ipython
----
-tags: [hide-output]
----
-!pip install interpolation
-```
-
 ## Overview
 
 In this lecture we continue the study of {doc}`the cake eating problem <cake_eating_problem>`.
@@ -47,9 +38,7 @@ We will use the following imports:
 
 ```{code-cell} ipython
 import matplotlib.pyplot as plt
-plt.rcParams["figure.figsize"] = (11, 5)  #set default figure size
 import numpy as np
-from interpolation import interp
 from scipy.optimize import minimize_scalar, bisect
 ```
 
@@ -211,7 +200,7 @@ class CakeEating:
         """
 
         u, β = self.u, self.β
-        v = lambda x: interp(self.x_grid, v_array, x)
+        v = lambda x: np.interp(x, self.x_grid, v_array)
 
         return u(c) + β * v(x - c)
 ```
@@ -533,7 +522,7 @@ class OptimalGrowth(CakeEating):
         """
 
         u, β, α = self.u, self.β, self.α
-        v = lambda x: interp(self.x_grid, v_array, x)
+        v = lambda x: np.interp(x, self.x_grid, v_array)
 
         return u(c) + β * v((x - c)**α)
 ```
@@ -609,7 +598,7 @@ def K(σ_array, ce):
     u_prime, β, x_grid = ce.u_prime, ce.β, ce.x_grid
     σ_new = np.empty_like(σ_array)
 
-    σ = lambda x: interp(x_grid, σ_array, x)
+    σ = lambda x: np.interp(x, x_grid, σ_array)
 
     def euler_diff(c, x):
         return u_prime(c) - β * u_prime(σ(x - c))

lectures/coleman_policy_iter.md

@@ -30,7 +30,6 @@ In addition to what's in Anaconda, this lecture will need the following librarie
 tags: [hide-output]
 ---
 !pip install quantecon
-!pip install interpolation
 ```
 
 ## Overview
@@ -62,9 +61,7 @@ Let's start with some imports:
 
 ```{code-cell} ipython
 import matplotlib.pyplot as plt
-plt.rcParams["figure.figsize"] = (11, 5)  #set default figure size
 import numpy as np
-from interpolation import interp
 from quantecon.optimize import brentq
 from numba import njit
 ```
@@ -301,7 +298,7 @@ def euler_diff(c, σ, y, og):
     f, f_prime, u_prime = og.f, og.f_prime, og.u_prime
 
     # First turn σ into a function via interpolation
-    σ_func = lambda x: interp(grid, σ, x)
+    σ_func = lambda x: np.interp(x, grid, σ)
 
     # Now set up the function we need to find the root of.
     vals = u_prime(σ_func(f(y - c) * shocks)) * f_prime(y - c) * shocks

lectures/egm_policy_iter.md

@@ -23,14 +23,6 @@ kernelspec:
 :depth: 2
 ```
 
-In addition to what's in Anaconda, this lecture will need the following libraries:
-
-```{code-cell} ipython
----
-tags: [hide-output]
----
-!pip install interpolation
-```
 
 ## Overview
 
@@ -51,9 +43,7 @@ Let's start with some standard imports:
 
 ```{code-cell} ipython
 import matplotlib.pyplot as plt
-plt.rcParams["figure.figsize"] = (11, 5)  #set default figure size
 import numpy as np
-from interpolation import interp
 from numba import njit
 ```
 
@@ -188,7 +178,7 @@ def K(σ_array, og):
     y = grid + σ_array  # y_i = k_i + c_i
 
     # Linear interpolation of policy using endogenous grid
-    σ = lambda x: interp(y, σ_array, x)
+    σ = lambda x: np.interp(x, y, σ_array)
 
     # Allocate memory for new consumption array
     c = np.empty_like(grid)

lectures/ifp.md

@@ -30,7 +30,6 @@ In addition to what's in Anaconda, this lecture will need the following librarie
 tags: [hide-output]
 ---
 !pip install quantecon
-!pip install interpolation
 ```
 
 ## Overview
@@ -60,10 +59,8 @@ We'll need the following imports:
 
 ```{code-cell} ipython
 import matplotlib.pyplot as plt
-plt.rcParams["figure.figsize"] = (11, 5)  #set default figure size
 import numpy as np
 from quantecon.optimize import brentq
-from interpolation import interp
 from numba import njit, float64
 from numba.experimental import jitclass
 from quantecon import MarkovChain
@@ -437,7 +434,7 @@ def euler_diff(c, a, z, σ_vals, ifp):
 
     # Convert policy into a function by linear interpolation
     def σ(a, z):
-        return interp(asset_grid, σ_vals[:, z], a)
+        return np.interp(a, asset_grid, σ_vals[:, z])
 
     # Calculate the expectation conditional on current z
     expect = 0.0
@@ -663,7 +660,7 @@ def compute_asset_series(ifp, T=500_000, seed=1234):
 
     # Solve for the optimal policy
     σ_star = solve_model_time_iter(ifp, σ_init, verbose=False)
-    σ = lambda a, z: interp(ifp.asset_grid, σ_star[:, z], a)
+    σ = lambda a, z: np.interp(a, ifp.asset_grid, σ_star[:, z])
 
     # Simulate the exogeneous state process
     mc = MarkovChain(P)

lectures/ifp_advanced.md

@@ -30,7 +30,6 @@ In addition to what's in Anaconda, this lecture will need the following librarie
 tags: [hide-output]
 ---
 !pip install quantecon
-!pip install interpolation
 ```
 
 ## Overview
@@ -57,9 +56,7 @@ We require the following imports:
 
 ```{code-cell} ipython
 import matplotlib.pyplot as plt
-plt.rcParams["figure.figsize"] = (11, 5)  #set default figure size
 import numpy as np
-from interpolation import interp
 from numba import njit, float64
 from numba.experimental import jitclass
 from quantecon import MarkovChain
@@ -436,7 +433,7 @@ def K(a_in, σ_in, ifp):
     n = len(P)
 
     # Create consumption function by linear interpolation
-    σ = lambda a, z: interp(a_in[:, z], σ_in[:, z], a)
+    σ = lambda a, z: np.interp(a, a_in[:, z], σ_in[:, z])
 
     # Allocate memory
     σ_out = np.empty_like(σ_in)
@@ -636,7 +633,7 @@ def compute_asset_series(ifp, a_star, σ_star, z_seq, T=500_000):
     """
 
     # Create consumption function by linear interpolation
-    σ = lambda a, z: interp(a_star[:, z], σ_star[:, z], a)
+    σ = lambda a, z: np.interp(a, a_star[:, z], σ_star[:, z])
 
     # Simulate the asset path
     a = np.zeros(T+1)