8000 MAINT: Simplify block implementation by j-towns · Pull Request #9667 · numpy/numpy · GitHub
[go: up one dir, main page]
Skip to content

MAINT: Simplify block implementation #9667

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 23 commits into from
Nov 12, 2017
Merged

MAINT: Simplify block implementation #9667

Changes from 1 commit
Commits
Show all changes
23 commits
Select commit Hold shift + click to select a range
a5c6f0d
Simplify block implementation
j-towns Sep 7, 2017
2dcc9aa
np.block style improvements
j-towns Sep 7, 2017
e787a9f
Reflect asanyarray behaviour in block
j-towns Sep 8, 2017
95adb77
Add empty list comment to block depth check
j-towns Sep 10, 2017
07a3f43
Use strict type checking (not isinstance)
j-towns Sep 11, 2017
19fc68c
Re-add tuple type-check comment
j-towns Sep 11, 2017
997ac2c
Re-add `atleast_nd` function.
j-towns Sep 11, 2017
7eb1044
Add detailed comment to _block_check_depths_match
j-towns Sep 18, 2017
ff7f726
Extend comments _block_check_depths_match
j-towns Sep 20, 2017
6ecd2b4
Try not recomputing list_ndim
j-towns Sep 23, 2017
1211b70
Slight simplification to logic
j-towns Sep 25, 2017
ffa6cf6
Add two tests for different arr_ndims
j-towns Sep 25, 2017
3ed6936
Simplify further - matching docstring logic
j-towns Sep 25, 2017
5f9f1fa
Rename list_ndim to max_depth
j-towns Sep 25, 2017
8a83a5f
rm extra line from near top of shape_base
j-towns Sep 25, 2017
5a0557a
Pre-calculate max array ndim
j-towns Oct 12, 2017
c2b5be5
Further slight simplifications
j-towns Oct 12, 2017
bd6729d
Update block docstrings
j-towns Oct 12, 2017
ad278f3
Fix python 3.4 sequence error
j-towns Oct 12, 2017
2c1734b
Correct empty list ndim
j-towns Oct 26, 2017
eaddf39
Avoid using zip(*...) syntax
j-towns Nov 1, 2017
a5cbc93
Use builtin next method
j-towns Nov 1, 2017
a691f2d
Rm unnecessary enumerate
j-towns Nov 9, 2017
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Simplify further - matching docstring logic
  • Loading branch information
@j-towns
j-towns committed Sep 25, 2017
commit 3ed6936b82cb444e5516bda705701245d4fa4b4d
2 changes: 1 addition & 1 deletion numpy/core/shape_base.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -436,7 +436,7 @@ def block_recursion(arrays, depth=0):
arrs = [block_recursion(arr, depth+1) for arr in arrays]
arr_ndim = max(arr.ndim for arr in arrs)
arrs = [atleast_nd(a, arr_ndim) for a in arrs]
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Not a huge fan of how atleast_nd gets called repeatedly, rather than just once for each array.
< 93B3 /option>
Copy link
Contributor Author
@j-towns j-towns Oct 9, 2017 •
edited
Loading

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

The simplest way to get around this would be to have _block_check_depths_match also find the max array ndim. I'll have a go at drafting that change and see how it looks and whether it improves performance.

return _nx.concatenate(arrs, axis=depth+max(0, arr_ndim-list_ndim))
return _nx.concatenate(arrs, axis=depth-list_ndim)
else:
# We've 'bottomed out' - arrays is either a scalar or an array
return atleast_nd(arrays, list_ndim)
Expand Down
0