Получение определенных индексов из сетки в numpy

#python

#python

Вопрос:

Я запускаю следующий код:

     import numpy as np
    Z = np.arange(9).reshape(3,3)
    print(Z[[0,1],[0,2]])
 

Z — это:

0 1 2
3 4 5
6 7 8

Я ожидал получить значения 1 и 2, но получил 0 и 5, поэтому я не понимаю, как работает эта индексация.

Спасибо

Ответ №1:

Z[[0,1], [0,2]] возвращает Z [0,0] и Z [1,2], поэтому вы получаете 0 и 5.

Комментарии:

1. спасибо, я думал, что каждая пара ссылается на другой элемент и не знал, что он распаковывается в этой форме.