Получить элемент многомерного массива numpy

@Sorg · Регистрация: 02.11.2021

Студворк — интернет-сервис помощи студентам

Есть n-мерный массив. Есть одномерный массив из n элементов, в нем лежат индексы элемента из первого массива. Нужно получить элемент исходного массива.
Например:

Python
1
2
3
4
5
a=np.arange(4).reshape(2,2)
array([[0, 1],
       [2, 3]])
b = np.arange(2)
array([0, 1])

В итоге надо получить a[b], т.е. элемент a[0,1].

@Gdez · 02.11.2021, 22:25

Sorg,

Python
1
a[np.split(b, b.size)][0]

@Sorg · 02.11.2021, 22:38 **[ТС]**

Вооот оно что!
Надо ему массив массивов закидывать в качестве индекса.
Спасибо!

@Sorg · 03.11.2021, 22:21 **[ТС]**

Кстати, нашел более кошерный вариант, если кому интересно

Python
1
a[tuple(b)]

@dondublon · 04.11.2021, 10:03

Сообщение от Sorg

Есть n-мерный массив.

Где? a у вас двумерный.

Сообщение от Sorg

a[tuple(b)]

При работе с numpy нежелательно переходить к нативным питоновским типам. Вам надо как-то разделить ваш индекс b на i и j (а лучше не делить, а сразу создавать i и j в виде массивов) и юзать так:

Python
1
a[i,j]

Понятно, если вам надо взять один элемент, это не имеет значения, но если много - уже сыграет.

@Sorg · 05.11.2021, 20:17 **[ТС]**

Где? a у вас двумерный.

В примере двумерный. Но нужен код, который работает в общем случае для n-мерного. В том числе и для двумерного, как пример.
Ваш код, конечно, хорош, но работает только для двумерного массива. Когда у меня строго двумерный, то я как раз таким кодом и пользуюсь.

@dondublon · 05.11.2021, 21:28

Sorg, индексации с произвольным числом измерений в numpy нет.
Что логично. Число измерений у чего-то в определённом месте программы должно быть фиксированным.

@Sorg · 06.11.2021, 21:37 **[ТС]**

Sorg, индексации с произвольным числом измерений в numpy нет.
Что логично. Число измерений у чего-то в определённом месте программы должно быть фиксированным.

Ну вот у меня numpy и произвольное число измерений и работает. ЧЯДНТ?

@dondublon · 06.11.2021, 21:50

Sorg, мне просто не приходило в голову делать тюплик из измерений. Я в своё время плотно работал с numpy. Когда у тебя тысячи (а то и десятки тысяч) индексов, и подобные операции повторяются, и должно работать быстро, как только возможно - как-то не приходит в голову делать из numpy-массивов тюплики.

Новые блоги и статьи Все статьи Все блоги /
изучаю kubernetes lagorue 13.01.2026 А пригодятся-ли мне знания kubernetes в России?	сукцессия микоризы: основная теория в виде двух уравнений. anaschu 11.01.2026 https:/ / rutube. ru/ video/ 7a537f578d808e67a3c6fd818a44a5c4/	WordPad для Windows 11 Jel 10.01.2026 WordPad для Windows 11 — это приложение, которое восстанавливает классический текстовый редактор WordPad в операционной системе Windows 11. После того как Microsoft исключила WordPad из. . .	Classic Notepad for Windows 11 Jel 10.01.2026 Old Classic Notepad for Windows 11 Приложение для Windows 11, позволяющее пользователям вернуть классическую версию текстового редактора «Блокнот» из Windows 10. Программа предоставляет более. . .
Почему дизайн решает? Neotwalker 09.01.2026 В современном мире, где конкуренция за внимание потребителя достигла пика, дизайн становится мощным инструментом для успеха бренда. Это не просто красивый внешний вид продукта или сайта — это. . .	Модель микоризы: классовый агентный подход 3 anaschu 06.01.2026 aa0a7f55b50dd51c5ec569d2d10c54f6/ O1rJuneU_ls https:/ / vkvideo. ru/ video-115721503_456239114	Owen Logic: О недопустимости использования связки «аналоговый ПИД» + RegKZR ФедосеевПавел 06.01.2026 Owen Logic: О недопустимости использования связки «аналоговый ПИД» + RegKZR ВВЕДЕНИЕ Введу сокращения: аналоговый ПИД — ПИД регулятор с управляющим выходом в виде числа в диапазоне от 0% до. . .	Модель микоризы: классовый агентный подход 2 anaschu 06.01.2026 репозиторий https:/ / github. com/ shumilovas/ fungi ветка по-частям. коммит Create переделка под биомассу. txt вход sc, но sm считается внутри мицелия. кстати, обьем тоже должен там считаться. . . .

@Sorg 1 / 1 / 0 Регистрация: 02.11.2021 Сообщений: 5
	02.11.2021, 22:38 [ТС]
	Вооот оно что! Надо ему массив массивов закидывать в качестве индекса. Спасибо! 0

@dondublon 4652 / 2072 / 366 Регистрация: 17.03.2012 Сообщений: 10,182 Записей в блоге: 6
	05.11.2021, 21:28
	Sorg, индексации с произвольным числом измерений в numpy нет. Что логично. Число измерений у чего-то в определённом месте программы должно быть фиксированным. 0

@dondublon 4652 / 2072 / 366 Регистрация: 17.03.2012 Сообщений: 10,182 Записей в блоге: 6
	06.11.2021, 21:50
	Sorg, мне просто не приходило в голову делать тюплик из измерений. Я в своё время плотно работал с numpy. Когда у тебя тысячи (а то и десятки тысяч) индексов, и подобные операции повторяются, и должно работать быстро, как только возможно - как-то не приходит в голову делать из numpy-массивов тюплики. 0

Получить элемент многомерного массива numpy

Решение