numpy.polynomial.polynomial.polyvander3d#

polynomial.polynomial.polyvander3d(x, y, z, deg)[source]#

Pseudo-Vandermonde matrix of given degrees.

Returns the pseudo-Vandermonde matrix of degrees deg and sample points (x, y, z). If l, m, n are the given degrees in x, y, z, then The pseudo-Vandermonde matrix is defined by

\[V[..., (m+1)(n+1)i + (n+1)j + k] = x^i * y^j * z^k,\]

where 0 <= i <= l, 0 <= j <= m, and 0 <= j <= n. The leading indices of V index the points (x, y, z) and the last index encodes the powers of x, y, and z.

If V = polyvander3d(x, y, z, [xdeg, ydeg, zdeg]), then the columns of V correspond to the elements of a 3-D coefficient array c of shape (xdeg + 1, ydeg + 1, zdeg + 1) in the order

\[c_{000}, c_{001}, c_{002},... , c_{010}, c_{011}, c_{012},...\]

and np.dot(V, c.flat) and polyval3d(x, y, z, c) will be the same up to roundoff. This equivalence is useful both for least squares fitting and for the evaluation of a large number of 3-D polynomials of the same degrees and sample points.

Parameters:
x, y, zarray_like

Arrays of point coordinates, all of the same shape. The dtypes will be converted to either float64 or complex128 depending on whether any of the elements are complex. Scalars are converted to 1-D arrays.

deglist of ints

List of maximum degrees of the form [x_deg, y_deg, z_deg].

Returns:
vander3dndarray

The shape of the returned matrix is x.shape + (order,), where \(order = (deg[0]+1)*(deg([1]+1)*(deg[2]+1)\). The dtype will be the same as the converted x, y, and z.

Notes

New in version 1.7.0.

Examples

>>> from numpy.polynomial import polynomial as P
>>> x = np.asarray([-1, 2, 1])
>>> y = np.asarray([1, -2, -3])
>>> z = np.asarray([2, 2, 5])
>>> l, m, n = [2, 2, 1]
>>> deg = [l, m, n]
>>> V = P.polyvander3d(x=x, y=y, z=z, deg=deg)
>>> V
array([[  1.,   2.,   1.,   2.,   1.,   2.,  -1.,  -2.,  -1.,
         -2.,  -1.,  -2.,   1.,   2.,   1.,   2.,   1.,   2.],
       [  1.,   2.,  -2.,  -4.,   4.,   8.,   2.,   4.,  -4.,
         -8.,   8.,  16.,   4.,   8.,  -8., -16.,  16.,  32.],
       [  1.,   5.,  -3., -15.,   9.,  45.,   1.,   5.,  -3.,
        -15.,   9.,  45.,   1.,   5.,  -3., -15.,   9.,  45.]])

We can verify the columns for any 0 <= i <= l, 0 <= j <= m, and 0 <= k <= n

>>> i, j, k = 2, 1, 0
>>> V[:, (m+1)*(n+1)*i + (n+1)*j + k] == x**i * y**j * z**k
array([ True,  True,  True])