class indexed_array: def __getitem__(self, indices): # convert a simple index x[y] to a tuple for consistency if not isinstance(indices, tuple): indices = tuple(indices) # now handle the different dimensional cases