Numpy массив меняет форму при доступе с индексами - PullRequest
Новая
       4

Numpy массив меняет форму при доступе с индексами

0 голосов
/ 11 сентября 2018

у меня маленькая матрица A с размерами MxNxO

У меня большая матрица B с размерами KxMxNxP, с P> O

У меня есть векторный индекс индексов размерности Ox1

Я хочу сделать: 

B[1,:,:,ind] = A

Но, слева от моего уравнения 

 B[1,:,:,ind].shape

имеет размерность Ox1xMxN, и поэтому я не могу транслировать в него A (MxNxO).

Почему доступ к B таким образом изменяет размеры левой стороны? Как я могу легко достичь своей цели? Спасибо

1 Ответ

0 голосов
/ 11 сентября 2018

Есть особенность, если не ошибка, то, что когда срезы смешиваются в середине расширенной индексации, нарезанные измерения помещаются в конец.

Так, например: 

In [204]: B = np.zeros((2,3,4,5),int)
In [205]: ind=[0,1,2,3,4]
In [206]: B[1,:,:,ind].shape
Out[206]: (5, 3, 4)

3,4 измерения были помещены после ind, 5.

. Мы можем обойти это путем индексации сначала с 1, а затем с остальными: 

In [207]: B[1][:,:,ind].shape
Out[207]: (3, 4, 5)
In [208]: B[1][:,:,ind] = np.arange(3*4*5).reshape(3,4,5)
In [209]: B[1]
Out[209]: 
array([[[ 0,  1,  2,  3,  4],
        [ 5,  6,  7,  8,  9],
        [10, 11, 12, 13, 14],
        [15, 16, 17, 18, 19]],

       [[20, 21, 22, 23, 24],
        [25, 26, 27, 28, 29],
        [30, 31, 32, 33, 34],
        [35, 36, 37, 38, 39]],

       [[40, 41, 42, 43, 44],
        [45, 46, 47, 48, 49],
        [50, 51, 52, 53, 54],
        [55, 56, 57, 58, 59]]])

Этоработает только тогда, когда этот первый индекс является скалярным.Если бы это тоже был список (или массив), мы получили бы промежуточную копию и не смогли бы установить значение, подобное этому.

https://docs.scipy.org/doc/numpy-1.15.0/reference/arrays.indexing.html#combining-advanced-and-basic-indexing

Это происходит в других SOвопросы, хотя и не в последнее время.

странный результат при использовании как индексирования срезов, так и логического индексирования в трехмерном массиве

...