Search code examples
pythonnumpybisect

'only length-1 arrays can be converted to Python scalars' error


I have such Python code

import numpy as np
import matplotlib.pyplot as plt
import math
from scipy import optimize as opt

def func1(x):
    f1 = math.exp(x-2)+x**3-x
    return f1

solv1_bisect = opt.bisect(func1, -1.5, 1.5)

x1 = np.linspace(-1.5,1.5) 
y1 = func1(x1)
plt.plot(x1,y1,'r-')
plt.grid()

print('solv1_bisect = ', solv1_bisect)

and I've got the error message such as

TypeError: only length-1 arrays can be converted to Python scalars

Please help me to fix it, Thanks!


Solution

  • The problem is that you are using math.exp that expects a Python scalar, for example:

    >>> import numpy as np
    >>> import math
    >>> math.exp(np.arange(3))  
    
    Traceback (most recent call last):
      File "path", line 3331, in run_code
        exec(code_obj, self.user_global_ns, self.user_ns)
      File "<ipython-input-10-3ac3b9478cd5>", line 1, in <module>
        math.exp(np.arange(3))
    TypeError: only size-1 arrays can be converted to Python scalars
    

    Use np.exp instead:

    def func1(x):
        f1 = np.exp(x - 2) + x ** 3 - x
        return f1
    

    The difference between np.exp and math.exp is that math.exp works with Python's numbers (floats and integers) while np.exp can work with numpy arrays. In your code the argument x is a numpy array, hence the error.