diff options
Diffstat (limited to 'numpy/lib/function_base.py')
| -rw-r--r-- | numpy/lib/function_base.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/numpy/lib/function_base.py b/numpy/lib/function_base.py index 4af867233..c3cce05ed 100644 --- a/numpy/lib/function_base.py +++ b/numpy/lib/function_base.py @@ -375,7 +375,7 @@ def average(a, axis=None, weights=None, returned=False): n = add.reduce(a*w, axis) d = add.reduce(w, axis) elif wsh == (ash[axis],): - ni = ash[axis] + ni = len(ash) r = [newaxis]*ni r[axis] = slice(None, None, 1) w1 = eval("w["+repr(tuple(r))+"]*ones(ash, float)") |