用字典切片NumPy数组

Question

Is there a simple option to slice a NumPy array with the predefined dictionary of indices?

For example:

>> a = array([3, 9, 1, 5, 5])

and (fictitious) dictionary:

>> index_dict = {'all_except_first': (1:None), 'all_except_last': (None:-1)}

and then:

>> a[index_dict['all_except_first']]
>> array([9, 1, 5, 5])
>> a[index_dict['all_except_first']]
>> array([3, 9, 1, 5])

Sort of slicing with names and not with numbers.

Answer 1

Create slice s:

>>> index_dict = {'all_except_first': slice(1, None), 'all_except_last': slice(None, -1)}
>>>
>>> a[index_dict['all_except_first']]
array([9, 1, 5, 5])
>>> a[index_dict['all_except_last']]
array([3, 9, 1, 5])