diff options
| author | Travis Oliphant <oliphant@enthought.com> | 2007-03-26 22:13:28 +0000 |
|---|---|---|
| committer | Travis Oliphant <oliphant@enthought.com> | 2007-03-26 22:13:28 +0000 |
| commit | 2ff5b35b2ce3fd485cedf8d9345b25065b8bc94d (patch) | |
| tree | 5831b384477654a21df62cb8f3ca7d624f01c373 /numpy/lib/function_base.py | |
| parent | bb10e4031198e6578b5b6a1cfe314fd3b9972037 (diff) | |
| download | numpy-2ff5b35b2ce3fd485cedf8d9345b25065b8bc94d.tar.gz | |
Fix ticket #459
Diffstat (limited to 'numpy/lib/function_base.py')
| -rw-r--r-- | numpy/lib/function_base.py | 5 |
1 files changed, 4 insertions, 1 deletions
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): |