2016-02-08 5 views
1

Я пытаюсь получить доступ к значению в многомерном массиве numpy. Это можно легко сделать, когда вы знаете все, для Exemple:Получите значение в массиве numpy из индекса в переменной

T = numpy.arrange(9).reshape(3, 3) T[2, 2]

И это возвращает 8, который является то, что я хочу. Теперь предположим, что [2, 2] хранится в переменной index. Как я могу сделать, чтобы значение в T с индексом, хранящимся в index? Я хотел бы сделать T[index], но он возвращает последнюю строку дважды (довольно логично, но не то, что я хочу).

Спасибо!

+5

Попробуйте 'ind = tuple (2,2); x [ind] ' – hpaulj

+0

Спасибо, отлично работает! – clems4ever

ответ

1

Попытка

ind=tuple(2,2) 
x[ind] 

x[2,2] такое же, как x[(2,2)] который переведен на вызов метода: x.__getitem__((2,2)).

Некоторые функции numpy создают индекс как список или массив, а затем преобразуют его в tuple для использования в индексе.

Смежные вопросы