From 43c6a89128347928c5fe26f67ba2a0a022f00822 Mon Sep 17 00:00:00 2001 From: John Kirkham Date: Fri, 22 Jan 2016 17:31:13 -0500 Subject: BUG: Make sure that the `ord=0` case returns a float. --- numpy/linalg/linalg.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'numpy/linalg/linalg.py') diff --git a/numpy/linalg/linalg.py b/numpy/linalg/linalg.py index fe2031efb..f333bde47 100644 --- a/numpy/linalg/linalg.py +++ b/numpy/linalg/linalg.py @@ -2150,7 +2150,7 @@ def norm(x, ord=None, axis=None, keepdims=False): return abs(x).min(axis=axis, keepdims=keepdims) elif ord == 0: # Zero norm - return (x != 0).sum(axis=axis, keepdims=keepdims) + return (x != 0).astype(float).sum(axis=axis, keepdims=keepdims) elif ord == 1: # special case for speedup return add.reduce(abs(x), axis=axis, keepdims=keepdims) -- cgit v1.2.1