Advertisement
Guest User

Untitled

a guest
Jul 24th, 2019
54
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.16 KB | None | 0 0
  1. >>> from numpy import *
  2. >>> import itertools
  3. >>> array([x for x in itertools.product(arange(2), arange(3))])
  4. array([[0, 0],
  5. [0, 1],
  6. [0, 2],
  7. [1, 0],
  8. [1, 1],
  9. [1, 2]])
  10.  
  11. In [14]: a = arange(5)
  12.  
  13. In [18]: r = array(meshgrid(a,a))
  14.  
  15. In [20]: r.shape
  16. Out[20]: (2, 5, 5)
  17.  
  18. In [21]: r
  19. Out[21]:
  20. array([[[0, 1, 2, 3, 4],
  21. [0, 1, 2, 3, 4],
  22. [0, 1, 2, 3, 4],
  23. [0, 1, 2, 3, 4],
  24. [0, 1, 2, 3, 4]],
  25.  
  26. [[0, 0, 0, 0, 0],
  27. [1, 1, 1, 1, 1],
  28. [2, 2, 2, 2, 2],
  29. [3, 3, 3, 3, 3],
  30. [4, 4, 4, 4, 4]]])
  31.  
  32. In [3]: a = arange(5)
  33.  
  34. In [4]: a[:,newaxis] * a[newaxis,:]
  35. Out[4]:
  36. array([[ 0, 0, 0, 0, 0],
  37. [ 0, 1, 2, 3, 4],
  38. [ 0, 2, 4, 6, 8],
  39. [ 0, 3, 6, 9, 12],
  40. [ 0, 4, 8, 12, 16]])
  41.  
  42. In [6]: a = ones((2,5))
  43.  
  44. In [7]: a[:,newaxis] * a[newaxis,:]
  45. Out[7]:
  46. array([[[ 1., 1., 1., 1., 1.],
  47. [ 1., 1., 1., 1., 1.]],
  48.  
  49. [[ 1., 1., 1., 1., 1.],
  50. [ 1., 1., 1., 1., 1.]]])
  51.  
  52. In [8]: a.shape
  53. Out[8]: (2, 5)
  54.  
  55. In [9]: b = a[:,newaxis] * a[newaxis,:]
  56.  
  57. In [10]: b.shape
  58. Out[10]: (2, 2, 5)
  59.  
  60. xx = numpy.zeros(n) + x.reshape(n,1)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement