diff options
| author | Eric Wieser <wieser.eric@gmail.com> | 2019-03-11 22:42:04 -0700 |
|---|---|---|
| committer | Eric Wieser <wieser.eric@gmail.com> | 2019-03-12 20:38:15 -0700 |
| commit | fcea19a3dd586bbf9d62719de551ac75d3b4e17a (patch) | |
| tree | 22fd41d0166b1a7dbe887ef3b09db9d5d5ad3f72 /numpy/polynomial/polynomial.py | |
| parent | 1bb279ad4f25d987155106ee6f82ba7fc83ce5a0 (diff) | |
| download | numpy-fcea19a3dd586bbf9d62719de551ac75d3b4e17a.tar.gz | |
MAINT: Unify polynomial valnd functions
No point writing the same function 12 times, when you can write it once
Diffstat (limited to 'numpy/polynomial/polynomial.py')
| -rw-r--r-- | numpy/polynomial/polynomial.py | 28 |
1 files changed, 4 insertions, 24 deletions
diff --git a/numpy/polynomial/polynomial.py b/numpy/polynomial/polynomial.py index 2b6c9fd0a..9a50e02bc 100644 --- a/numpy/polynomial/polynomial.py +++ b/numpy/polynomial/polynomial.py @@ -910,14 +910,7 @@ def polyval2d(x, y, c): .. versionadded:: 1.7.0 """ - try: - x, y = np.array((x, y), copy=0) - except Exception: - raise ValueError('x, y are incompatible') - - c = polyval(x, c) - c = polyval(y, c, tensor=False) - return c + return pu._valnd(polyval, c, x, y) def polygrid2d(x, y, c): @@ -970,9 +963,7 @@ def polygrid2d(x, y, c): .. versionadded:: 1.7.0 """ - c = polyval(x, c) - c = polyval(y, c) - return c + return pu._gridnd(polyval, c, x, y) def polyval3d(x, y, z, c): @@ -1023,15 +1014,7 @@ def polyval3d(x, y, z, c): .. versionadded:: 1.7.0 """ - try: - x, y, z = np.array((x, y, z), copy=0) - except Exception: - raise ValueError('x, y, z are incompatible') - - c = polyval(x, c) - c = polyval(y, c, tensor=False) - c = polyval(z, c, tensor=False) - return c + return pu._valnd(polyval, c, x, y, z) def polygrid3d(x, y, z, c): @@ -1087,10 +1070,7 @@ def polygrid3d(x, y, z, c): .. versionadded:: 1.7.0 """ - c = polyval(x, c) - c = polyval(y, c) - c = polyval(z, c) - return c + return pu._gridnd(polyval, c, x, y, z) def polyvander(x, deg): |