Advertisement
Guest User

Untitled

a guest
Jun 19th, 2019
72
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.63 KB | None | 0 0
  1. u = np.array([[100., 100., 100., 100., 100.],
  2. [100., 0., 0., 0., 100.],
  3. [100., 0., 0., 0., 100.],
  4. [100., 0., 0., 0., 100.],
  5. [100., 100., 100., 100., 100.]])
  6.  
  7. [[100. 100. 100. 100. 100.]
  8. [100. 50. 25. 50. 100.]
  9. [100. 25. 0. 25. 100.]
  10. [100. 50. 25. 50. 100.]
  11. [100. 100. 100. 100. 100.]]
  12.  
  13. import numpy as np
  14.  
  15. #example array
  16. x = np.array([[100., 2., 3., 7., 100.],
  17. [100., 5., 3., 7., 100.],
  18. [100., 3., 6., 3., 100.],
  19. [50., 4., 5., 2., 100.],
  20. [100., 100., 100., 100., 100.]])
  21.  
  22. lbarrier = (np.size(x, 1))+1
  23. rbarrier = (np.size(x, 1))-1
  24. bbarrier = (np.size(x, 0))-1
  25.  
  26.  
  27.  
  28.  
  29. #inner vals
  30. itops = x[0:bbarrier][1:rbarrier][0][1:rbarrier]
  31. ibot = x[0:bbarrier][1:rbarrier][-1][1:rbarrier]
  32. ileft = x[1:bbarrier,1]
  33. iright = x[1:bbarrier,-2]
  34.  
  35.  
  36.  
  37. #edge vals
  38. etops = np.array(x[0:][0:][0][1:-1])
  39. ebot = np.array(x[0:][0:][-1][1:-1])
  40. eleft = np.array(x[1:bbarrier, 0])
  41. eright = np.array(x[1:bbarrier, -1])
  42.  
  43. In [49]: from scipy import ndimage
  44.  
  45. In [50]: import numpy as np
  46.  
  47. In [51]: k
  48. Out[51]:
  49. array([[0. , 0.25, 0. ],
  50. [0.25, 0. , 0.25],
  51. [0. , 0.25, 0. ]])
  52.  
  53. In [52]: u = np.zeros((3,3))
  54.  
  55. In [53]: ndimage.convolve(u, k, mode='constant', cval=100.0)
  56. Out[53]:
  57. array([[50., 25., 50.],
  58. [25., 0., 25.],
  59. [50., 25., 50.]])
  60.  
  61. In [54]:
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement