diff options
| author | Nikita Titov <nekit94-08@mail.ru> | 2018-08-26 19:15:27 +0300 |
|---|---|---|
| committer | Charles Harris <charlesr.harris@gmail.com> | 2018-08-26 11:15:27 -0500 |
| commit | 63a57c3f6a58c61763daf89a2d5495f6a855bf9b (patch) | |
| tree | c46fc415a0bb045874399fb4d6f49af8a3fee489 /numpy/lib/shape_base.py | |
| parent | ab30683fe1277a7fd471be63975a02ed8d766f94 (diff) | |
| download | numpy-63a57c3f6a58c61763daf89a2d5495f6a855bf9b.tar.gz | |
BUG: fix array_split incorrect behavior with array size bigger MAX_INT32 (#11813)
Fixes #11809.
* BUG: fix array_split incorrect behavior with array size bigger MAX_INT32
* TST: added test for array_split with array size greater MAX_INT32
* addressed review comments
Diffstat (limited to 'numpy/lib/shape_base.py')
| -rw-r--r-- | numpy/lib/shape_base.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/numpy/lib/shape_base.py b/numpy/lib/shape_base.py index d31d8a939..66f534734 100644 --- a/numpy/lib/shape_base.py +++ b/numpy/lib/shape_base.py @@ -688,7 +688,7 @@ def array_split(ary, indices_or_sections, axis=0): except AttributeError: Ntotal = len(ary) try: - # handle scalar case. + # handle array case. Nsections = len(indices_or_sections) + 1 div_points = [0] + list(indices_or_sections) + [Ntotal] except TypeError: @@ -700,7 +700,7 @@ def array_split(ary, indices_or_sections, axis=0): section_sizes = ([0] + extras * [Neach_section+1] + (Nsections-extras) * [Neach_section]) - div_points = _nx.array(section_sizes).cumsum() + div_points = _nx.array(section_sizes, dtype=_nx.intp).cumsum() sub_arys = [] sary = _nx.swapaxes(ary, axis, 0) |