Evaluate a polynomial at points x and the shape of the coefficient array extended for each dimension of x in Python

To evaluate a polynomial at points x, use the polynomial.polyval() method in Python Numpy. The 1st parameter, x, if x is a list or tuple, it is converted to an ndarray, otherwise it is left unchanged and treated as a scalar. In either case, x or its elements must support addition and multiplication with themselves and with the elements of c.

The 2nd parameter, C, an array of coefficients ordered so that the coefficients for terms of degree n are contained in c[n]. If c is multidimensional the remaining indices enumerate multiple polynomials. In the two dimensional case the coefficients may be thought of as stored in the columns of c.

The 3rd parameter, tensor, if True, the shape of the coefficient array is extended with ones on the right, one for each dimension of x. Scalars have dimension 0 for this action. The result is that every column of coefficients in c is evaluated for every element of x. If False, x is broadcast over the columns of c for the evaluation. This keyword is useful when c is multidimensional. The default value is True.

Steps

At first, import the required libraries −

import numpy as np
from numpy.polynomial.polynomial import polyval

Create a multidimensional array of coefficients −

c = np.arange(4).reshape(2,2)

Display the array −

print("Our Array...\n",c)

Check the Dimensions −

print("\nDimensions of our Array...\n",c.ndim)

Get the Datatype −

print("\nDatatype of our Array object...\n",c.dtype)

Get the Shape −

print("\nShape of our Array object...\n",c.shape)

To evaluate a polynomial at points x, use the polynomial.polyval() method in Python Numpy −

print("\nResult...\n",polyval([1,2], c, tensor=True))

Example

import numpy as np
from numpy.polynomial.polynomial import polyval

# Create a multidimensional array of coefficients
c = np.arange(4).reshape(2,2)

# Display the array
print("Our Array...\n",c)

# Check the Dimensions
print("\nDimensions of our Array...\n",c.ndim)

# Get the Datatype
print("\nDatatype of our Array object...\n",c.dtype)

# Get the Shape
print("\nShape of our Array object...\n",c.shape)

# To evaluate a polynomial at points x, use the polynomial.polyval() method in Python Numpy
print("\nResult...\n",polyval([1,2], c, tensor=True))

Output

Our Array...
[[0 1]
   [2 3]]

Dimensions of our Array...
2

Datatype of our Array object...
int64

Shape of our Array object...
(2, 2)

Result...
[[2. 4.]
   [4. 7.]]