From 2ff5b35b2ce3fd485cedf8d9345b25065b8bc94d Mon Sep 17 00:00:00 2001 From: Travis Oliphant Date: Mon, 26 Mar 2007 22:13:28 +0000 Subject: Fix ticket #459 --- numpy/lib/function_base.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) (limited to 'numpy/lib') diff --git a/numpy/lib/function_base.py b/numpy/lib/function_base.py index 8f08a73a4..163a129f7 100644 --- a/numpy/lib/function_base.py +++ b/numpy/lib/function_base.py @@ -422,7 +422,10 @@ def select(condlist, choicelist, default=0): pfac = asarray(1) for k in range(n2+1): pfac = pfac + asarray(choicelist[k]) - S = S*ones(asarray(pfac).shape) + if type(S) in ScalarType: + S = S*ones(asarray(pfac).shape, type(S)) + else: + S = S*ones(asarray(pfac).shape, S.dtype) return choose(S, tuple(choicelist)) def _asarray1d(arr, copy=False): -- cgit v1.2.1