neighbours = [ grid[ i, j, 0 ], 
                           grid[ i, j, 2 ], 
                           np.roll( grid, -1, axis=0 )[ i, j, 0 ] ]