diff options
| author | Eric Wieser <wieser.eric@gmail.com> | 2019-08-24 14:29:42 -0700 |
|---|---|---|
| committer | Eric Wieser <wieser.eric@gmail.com> | 2019-09-14 19:55:48 -0700 |
| commit | e443198415038a5995a15c6c32b6de0919c95f70 (patch) | |
| tree | d0c063724c5c7c872b725f305c37549b62be71f0 /numpy/polynomial/legendre.py | |
| parent | 99c62dc44a70af407f266eb30b756b297c215899 (diff) | |
| download | numpy-e443198415038a5995a15c6c32b6de0919c95f70.tar.gz | |
MAINT: polynomial: Add an N-d vander implementation used under the hood of the vander2d and vander3d functions
The generalization is not exposed in the public API yet, but it could be if the need arises.
The shape / dtype conversion logic is left as is for now, even if it might be broken.
Diffstat (limited to 'numpy/polynomial/legendre.py')
| -rw-r--r-- | numpy/polynomial/legendre.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/numpy/polynomial/legendre.py b/numpy/polynomial/legendre.py index 281982d0b..c11824761 100644 --- a/numpy/polynomial/legendre.py +++ b/numpy/polynomial/legendre.py @@ -1229,7 +1229,7 @@ def legvander2d(x, y, deg): .. versionadded:: 1.7.0 """ - return pu._vander2d(legvander, x, y, deg) + return pu._vander_nd_flat((legvander, legvander), (x, y), deg) def legvander3d(x, y, z, deg): @@ -1283,7 +1283,7 @@ def legvander3d(x, y, z, deg): .. versionadded:: 1.7.0 """ - return pu._vander3d(legvander, x, y, z, deg) + return pu._vander_nd_flat((legvander, legvander, legvander), (x, y, z), deg) def legfit(x, y, deg, rcond=None, full=False, w=None): |