Найти индекс максимального элемент двумерного массива

@orAnd · Регистрация: 13.04.2014

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

Задача такая: есть двумерный массив типа:

Python
1
[[1.0, 2.0, 1.5], [2.0, 0.5, 2.0], [1.0, 1.1, -0.1]]

надо найти индексы всех максимальных элементов то есть должен вернуть:

Python
1
[[0,1],[1,0],[1,2]]

Есть ли какое ниудь изящное решение?
Код:

Python
1
2
3
4
5
6
7
8
9
sk=[[1.0, 2.0, 1.5], [2.0, 0.5, 2.0], [1.0, 1.1, -0.1]]
b=3
m = max([e for elem in sk for e in elem])# Максимальный элемент
l = []
for x in range(b):#массивы "квадратные" со стороной b
    for y in range(b):
        if sk[x][y] == m:
            l.append([x, y])
print(l)

@Somebody · 07.09.2014, 17:34

Python
1
2
3
[(x, y) for x in range(b) for y in range(b) if sk[x][y] == m]
# или
[(x, y) for x, col in enumerate(sk) for y, n in enumerate(col) if n == m]

@dondublon · 09.09.2014, 11:23

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

надо найти индексы всех максимальных элементов то есть должен вернуть:

Насколько я понял задачу - найти ИНДЕКСЫ и ВСЕХ максимальных элементов, то вернуть надо:
[[1], [0, 2], [1]]

В качестве изящного решения - использовать numpy и функцию argsorted. Правда, она возвращает индексы не всех, а просто упорядочивает, но если надо именно всех, то можно скомбинировать.

Python
1
2
arr = np.array([[1.0, 2.0, 1.5], [2.0, 0.5, 2.0], [1.0, 1.1, -0.1]])
idx1 = np.argsort(arr, axis=1)

результат:

Code
1
2
3
array([[0, 2, 1],
       [1, 0, 2],
       [2, 0, 1]], dtype=int64)

Индексы максимальных элементов по строкам собраны в последнем столбце idx1.

Python
1
2
3
4
idxmax = idx1[:,-1]
# Сами элементы, думал, как сделать в одну операцию, векторно, но не получилось.
for i in xrange(3):
    print arr[i][idxmax[i]]

Добавлено через 1 час 20 минут
По поводу последнего - взять элементы по индексам без цикла можно так:

Python
1
arr[basic, idxmax]

где

Python
1
basic = range(3) или np.arange(3)

Новые блоги и статьи Все статьи Все блоги /
делаю науч статью по влиянию грибов на сукцессию anaschu 13.03.2026 прикрепляю статью	SDL3 для Desktop (MinGW): Создаём пустое окно с нуля для 2D-графики на SDL3, Си и C++ 8Observer8 10.03.2026 Содержание блога Финальные проекты на Си и на C++: hello-sdl3-c. zip hello-sdl3-cpp. zip Результат:	Установка CMake и MinGW 13.1 для сборки С и C++ приложений из консоли и из Qt Creator в EXE 8Observer8 10.03.2026 Содержание блога MinGW - это коллекция инструментов для сборки приложений в EXE. CMake - это система сборки приложений. Здесь описаны базовые шаги для старта программирования с помощью CMake и. . .	Как дизайн сайта влияет на конверсию: 7 решений, которые реально повышают заявки Neotwalker 08.03.2026 Многие до сих пор воспринимают дизайн сайта как “красивую оболочку”. На практике всё иначе: дизайн напрямую влияет на то, оставит человек заявку или уйдёт через несколько секунд. Даже если у вас. . .
Модульная разработка через nuget packages DevAlt 07.03.2026 Сложившийся в . Net-среде способ разработки чаще всего предполагает монорепозиторий в котором находятся все исходники. При создании нового решения, мы просто добавляем нужные проекты и имеем. . .	Модульный подход на примере F# DevAlt 06.03.2026 В блоге дяди Боба наткнулся на такое определение: В этой книге («Подход, основанный на вариантах использования») Ивар утверждает, что архитектура программного обеспечения — это структуры,. . .	Управление камерой с помощью скрипта 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 На первой гифке отладочные линии отключены, а на второй включены:. . .