Получить элемент многомерного массива 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-массивов тюплики.

Новые блоги и статьи Все статьи Все блоги /
Управление камерой с помощью скрипта OrbitControls.js на Three.js: Вращение, зум и панорамирование 8Observer8 05.03.2026 Содержание блога Финальная демка в браузере работает на Desktop и мобильных браузерах. Итоговый код: orbit-controls-threejs-js. zip. Сканируйте QR-код на мобильном. Вращайте камеру одним пальцем,. . .	SDL3 для Web (WebAssembly): Синхронизация спрайтов SDL3 и тел Box2D 8Observer8 04.03.2026 Содержание блога Финальная демка в браузере. Итоговый код: finish-sync-physics-sprites-sdl3-c. zip На первой гифке отладочные линии отключены, а на второй включены:. . .	SDL3 для Web (WebAssembly): Идентификация объектов на Box2D v3 - использование userData и событий коллизий 8Observer8 02.03.2026 Содержание блога Финальная демка в браузере. Итоговый код: finish-collision-events-sdl3-c. zip Сканируйте QR-код на мобильном и вы увидите, что появится джойстик для управления главным героем. . . .	Реалии Hrethgir 01.03.2026 Нет, я не закончил до сих пор симулятор. Эта задача сложнее. Не получилось уйти в плавсостав, но оно и к лучшему, возможно. Точнее получалось - но сварщиком в палубную команду, а это значит, в моём. . .
Ритм жизни kumehtar 27.02.2026 Иногда приходится жить в ритме, где дел становится всё больше, а вовлечения в происходящее — всё меньше. Плотный график не даёт вниманию закрепиться ни на одном событии. Утро начинается с быстрых,. . .	SDL3 для Web (WebAssembly): Сборка библиотек: SDL3, Box2D, FreeType, SDL3_ttf, SDL3_mixer и SDL3_image из исходников с помощью CMake и Emscripten 8Observer8 27.02.2026 Недавно вышла версия 3. 4. 2 библиотеки SDL3. На странице официальной релиза доступны исходники, готовые DLL (для x86, x64, arm64), а также библиотеки для разработки под Android, MinGW и Visual Studio. . . .	SDL3 для Web (WebAssembly): Реализация движения на Box2D v3 - трение и коллизии с повёрнутыми стенами 8Observer8 20.02.2026 Содержание блога Box2D позволяет легко создать главного героя, который не проходит сквозь стены и перемещается с заданным трением о препятствия, которые можно располагать под углом, как верхнее. . .	Конвертировать закладки radiotray-ng в m3u-плейлист damix 19.02.2026 Это можно сделать скриптом для PowerShell. Использование . \СonvertRadiotrayToM3U. ps1 <path_to_bookmarks. json> Рядом с файлом bookmarks. json появится файл bookmarks. m3u с результатом. # Check if. . .

@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

Решение