numpy - двумерное индексирование массива - PullRequest
5 голосов
/ 03 марта 2011

В соответствии с документами поведение numpy по умолчанию заключается в индексации массивов сначала по строкам, а затем по столбцам:

a = numpy.arange(6).reshape(3,2)

[[0 1]
 [2 3]
 [4 5]]

print a[0][1] # is 1

Я хочу проиндексировать массив, используя геометрически ориентированное соглашение a[x][y], как по осям X и Y. Как изменить порядок индексации без изменения формы массива, чтобы a[0][1] вернул 2?

1 Ответ

8 голосов
/ 03 марта 2011

Вы можете написать a.T[0,1], чтобы использовать индексы транспонирования массива, наоборот в 2D.

...