diff options
Diffstat (limited to 'numpy/lib/function_base.py')
| -rw-r--r-- | numpy/lib/function_base.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/numpy/lib/function_base.py b/numpy/lib/function_base.py index 80eaf8acf..0dee7784a 100644 --- a/numpy/lib/function_base.py +++ b/numpy/lib/function_base.py @@ -657,11 +657,11 @@ def select(condlist, choicelist, default=0): Examples -------- - >>> x = np.arange(10) - >>> condlist = [x<3, x>5] + >>> x = np.arange(6) + >>> condlist = [x<3, x>3] >>> choicelist = [x, x**2] >>> np.select(condlist, choicelist) - array([ 0, 1, 2, ..., 49, 64, 81]) + array([ 0, 1, 2, 0, 16, 25]) """ # Check the size of condlist and choicelist are the same, or abort. |