Python - Method to get the array around an slice in a matrix