在Python中，有没有一种方法可以保存数组的索引子集以供以后使用？

Question

My code currently has an array, lets say for example:

arr = np.ones((512, 512)).

There is an area of the array that interests me. I usually access it like this:

arr[50:200,150:350] #do stuff here.

I was wondering, is there some way to make a variable that holds [50:200,150:350] ? This way, if I need to slightly change my mask, I can do it once, on the top of the file, instead of everywhere it is accessed.

I tried mask = [50:200,150:350] , arr[mask] but Python syntax won't allow that.

Thanks for the help!

Answer 1

Apparently numpy extends slicing and allows multiple slice() objects, one per dimension.

import numpy
o = numpy.ones((32, 32))
print(o[3:5,3:5])

foo = slice(3,5), slice(3,5)
print(o[foo])

Both incantations produce same result :)