Странное поведение индексирования кортежа массивом numpy

Question

Я заметил некоторое запутанное поведение при индексировании плоского массива numpy со списком кортежей (с использованием python 2.7.8 и numpy 1.9.1). Я предполагаю, что это связано с максимальным количеством измерений массива (я считаю, это 32), но я не смог найти документацию.

>>> a = np.arange(100) 
>>> tuple_index = [(i,) for i in a] 
>>> a[tuple_index] # This works (but maybe it shouldn't) 
>>> a[tuple_index[:32]] # This works too 
>>> a[tuple_index[:31]] # This breaks for 2 <= i < 32 
Traceback (most recent call last): 
    File "<stdin>", line 1, in <module> 
IndexError: too many indices for array 
>>> a[tuple_index[:1]] # This also works... 

Является ли список кортежей «сплющенным», если он составляет 32 элемента или больше? Это где-то документально?

Answer 1

Разница заключается в том, что первые примеры запускают причудливое индексирование (которое просто выбирает индексы в списке из того же размера), тогда как tuple_index[:31] вместо этого рассматривается как индексирующий кортеж (что подразумевает выбор из нескольких осей).

Как отмечался выше, максимальное количество измерений для массива NumPy (обычно) 32:

>>> np.MAXDIMS 
32 

Согласно следующему комментарию в файле mapping.c (который содержит код для интерпретации индекса, принятого пользователь), любая последовательность кортежей короче 32 сплющивается к индексации кортежа:.

/* 
* Sequences < NPY_MAXDIMS with any slice objects 
* or newaxis, Ellipsis or other arrays or sequences 
* embedded, are considered equivalent to an indexing 
* tuple. (`a[[[1,2], [3,4]]] == a[[1,2], [3,4]]`) 
*/ 

(я еще не нашел ссылку на это в официальной документации на сайте SciPy)

Это делает a[tuple_index[:3]] эквивалентом a[(0,), (1,), (2,)], поэтому ошибка «слишком много индексов» (потому что a имеет только одно измерение, но мы подразумеваем, что есть три).

С другой стороны, a[tuple_index] - это то же самое, что и a[[(0,), (1,), (2,), ..., (99,)]], что приводит к 2D-массиву.