Guest User

Untitled

a guest
Mar 20th, 2018
117
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.47 KB | None | 0 0
  1. np.stack(a, axis=0)
  2.  
  3. >>> import numpy as np
  4. >>> a = np.empty(3, dtype=object)
  5. >>> a[0] = np.array([[1], [2]])
  6. >>> a[1] = np.array([[3], [4]])
  7. >>> a[2] = np.array([[5], [6]])
  8. >>> a
  9. array([array([[1],
  10. [2]]), array([[3],
  11. [4]]),
  12. array([[5],
  13. [6]])], dtype=object)
  14. >>> a.shape
  15. (3,)
  16. >>> a[0].shape
  17. (2, 1)
  18. >>> np.stack(a, axis=0)
  19. array([[[1],
  20. [2]],
  21.  
  22. [[3],
  23. [4]],
  24.  
  25. [[5],
  26. [6]]])
  27. >>> np.stack(a, axis=0).shape
  28. (3, 2, 1)
Add Comment
Please, Sign In to add comment