Как правильно индексировать массивы numpy с помощью массивов (или чего-либо еще, что поддерживает сложение, чтобы его можно было сместить) [дубликат]

На этот вопрос уже есть ответы здесь :

Подмножество двумерного массива numpy(4 ответа )

Закрыто 4 года назад,

Самый простой способ объяснить мой вопрос может быть на примере, поэтому позвольте мне определить некоторые массивы:

>>> test = arange(25).reshape((5,5))
>>> test
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]])
>>> Xinds = array([1,2,3])
>>> Yinds = array([1,2,3])

Теперь, если бы я хотел элементы в строках 1, 2 и 3 и в столбце 0, я мог бы пойти:

>>> test[Yinds,0]
array([ 5, 10, 15])

Если бы мне нужны были элементы в строках 1, 2 и 3 и во всех столбцах, я мог бы пойти:

>>> test[Yinds, :]
array([[ 5,  6,  7,  8,  9],
       [10, 11, 12, 13, 14],
       [15, 16, 17, 18, 19]])

Однако, если я попытаюсь расширить это чтобы получить элементы в строках 1, 2 и 3 и в столбцах 1, 2 и 3, -- сюрприз! -- вместо этого я получаю элементы в (1,1), (2,2) и (3,3 )

>>> test[Yinds, Xinds]
array([ 6, 12, 18])

Вместо того, что я хочу:

>>> test[Yinds, :][:, Xinds]
array([[ 6,  7,  8],
       [11, 12, 13],
       [16, 17, 18]])
>>> test[1:4,1:4]
array([[ 6,  7,  8],
       [11, 12, 13],
       [16, 17, 18]])

Я понимаю, что могу определить срез, но я хочу иметь возможность добавлять смещение к индексам (например, Yinds+offset), а это нельзя сделать с помощью срезов.

Я мог бы сделать что-то вроде

>>> xStart = 1
>>> xEnd   = 4
>>> yStart = 1
>>> yEnd   = 4

>>> offset = 1
>>> test[yStart+offset:yEnd+offset, xStart+offset:xEnd+offset]
...

или

>>> Xinds = array([1,4])
>>> Yinds = array([1,4])

>>> offset = 1
>>> test[slice(*(Yinds+offset)), slice(*(Xinds+offset))]
...

Но ни один из них не является особенно чистым

Обезьяна, исправляющая оператор сложения в срезе, кажется, не вариант, и наследование от слайса для добавления оператор тоже не работает; Я получаю сообщение об ошибке: "тип 'slice' не является приемлемым базовым типом".(*Ворчание* Это не было бы проблемой в Ruby *Ворчание*)

Итак, мой вопрос: как проще всего получить доступ к (более чем одномерному) подмассиву с чем-то, что можно хранить и смещать? ?

Варианты на данный момент:

test[Yinds+offset, :][:, Xinds+offset]
test[yStart+offset:yEnd+offset, xStart+offset:xEnd+offset]
test [slice(*(Yinds+offset)), slice(*(Xinds+offset))]

задан lnmaurer 12 June 2012 в 01:15

0 ответов

Другие вопросы по тегам:

indexing numpy offset python

Как правильно индексировать массивы numpy с помощью массивов (или чего-либо еще, что поддерживает сложение, чтобы его можно было сместить) [дубликат]

0 ответов

Похожие вопросы: