diff options
author | Eric Wieser <wieser.eric@gmail.com> | 2018-09-17 20:59:19 -0700 |
---|---|---|
committer | Eric Wieser <wieser.eric@gmail.com> | 2018-09-17 21:10:28 -0700 |
commit | 9904fc95f0917af3f2f811aadd1ebdafca95d876 (patch) | |
tree | d494a205bff1da29febd0783b4d4b64cc5f5f28f /numpy/core/shape_base.py | |
parent | b5f56572ac656f95c8ecc19bf506983e9df78df9 (diff) | |
download | numpy-9904fc95f0917af3f2f811aadd1ebdafca95d876.tar.gz |
DOC: Use a standard-style docstring to explain `_block_check_depths_match`
Diffstat (limited to 'numpy/core/shape_base.py')
-rw-r--r-- | numpy/core/shape_base.py | 25 |
1 files changed, 16 insertions, 9 deletions
diff --git a/numpy/core/shape_base.py b/numpy/core/shape_base.py index 3e023c87b..9297af816 100644 --- a/numpy/core/shape_base.py +++ b/numpy/core/shape_base.py @@ -371,15 +371,22 @@ def _block_check_depths_match(arrays, parent_index=[]): for each innermost list, in case an error needs to be raised, so that the index of the offending list can be printed as part of the error. - The parameter `parent_index` is the full index of `arrays` within the - nested lists passed to _block_check_depths_match at the top of the - recursion. - The return value is a pair. The first item returned is the full index - of an element (specifically the first element) from the bottom of the - nesting in `arrays`. An empty list at the bottom of the nesting is - represented by a `None` index. - The second item is the maximum of the ndims of the arrays nested in - `arrays`. + Parameters + ---------- + arrays : nested list of arrays + The arrays to check + parent_index : list of int + The full index of `arrays` within the nested lists passed to + `_block_check_depths_match` at the top of the recursion. + + Returns + ------- + first_index : list of int + The full index of the first element from the bottom of the nesting in + `arrays`. An empty list at the bottom of the nesting is represented by + a `None` index. + max_arr_ndim : int + The maximum of the ndims of the arrays nested in `arrays`. """ def format_index(index): idx_str = ''.join('[{}]'.format(i) for i in index if i is not None) |