Индексирование NumPy с помощью List?

Question

SOF,

Я заметил интересную NumPy демо в этом URL:

http://cs231n.github.io/python-numpy-tutorial/

Я вижу это:

import numpy as np 

a = np.array([[1,2], [3, 4], [5, 6]]) 

# An example of integer array indexing. 
# The returned array will have shape (3,) and 

print(a[[0, 1, 2], [0, 1, 0]]) 
# Prints "[1 4 5]" 

Я понимаю, с помощью целых чисел в качестве аргументов индекса:

a[1,1] 

и th имеет синтаксис:

a[0:2,:] 

Вообще, Если я использую список как синтаксис индекса, что это значит?

В частности, я не понимаю, почему:

print(a[[0, 1, 2], [0, 1, 0]]) 
# Prints "[1 4 5]" 

Answer 1

Последнее утверждение напечатает (в матричных обозначениях) a(0,0), a(1,1) и a(2,0). В питонных обозначениях это a[0][0], a[1][1] и a[2][0].

Первый индексный индекс содержит индексы для первой оси (обозначение матрицы: индекс строки), второй список содержит индексы для второй оси (индекс столбца).