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

@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)

Новые блоги и статьи Все статьи Все блоги /
Советы по крайней бережливости. Внимание, это ОЧЕНЬ длинный пост. Programma_Boinc 28.12.2025 Советы по крайней бережливости. Внимание, это ОЧЕНЬ длинный пост. Налог на собак: https:/ / ********/ gallery/ V06K53e Финансовый отчет в Excel: https:/ / ********/ gallery/ bKBkQFf Пост отсюда. . .	Кто-нибудь знает, где можно бесплатно получить настольный компьютер или ноутбук? США. Programma_Boinc 26.12.2025 Нашел на реддите интересную статью под названием Anyone know where to get a free Desktop or Laptop? Ниже её машинный перевод. После долгих разбирательств я наконец-то вернула себе. . .	Thinkpad X220 Tablet — это лучший бюджетный ноутбук для учёбы, точка. Programma_Boinc 23.12.2025 Рецензия / Мнение/ Перевод Нашел на реддите интересную статью под названием The Thinkpad X220 Tablet is the best budget school laptop period . Ниже её машинный перевод. Thinkpad X220 Tablet —. . .	PhpStorm 2025.3: WSL Terminal всегда стартует в ~ and_y87 14.12.2025 PhpStorm 2025. 3: WSL Terminal всегда стартует в ~ (home), игнорируя директорию проекта Симптом: После обновления до PhpStorm 2025. 3 встроенный терминал WSL открывается в домашней директории. . .	Как объединить две одинаковые БД Access с разными данными VikBal 11.12.2025 Помогите пожалуйста !! Как объединить 2 одинаковые БД Access с разными данными.
Новый ноутбук volvo 07.12.2025 Всем привет. По скидке в "черную пятницу" взял себе новый ноутбук Lenovo ThinkBook 16 G7 на Амазоне: Ryzen 5 7533HS 64 Gb DDR5 1Tb NVMe 16" Full HD Display Win11 Pro	Музыка, написанная Искусственным Интеллектом volvo 04.12.2025 Всем привет. Некоторое время назад меня заинтересовало, что уже умеет ИИ в плане написания музыки для песен, и, собственно, исполнения этих самых песен. Стихов у нас много, уже вышли 4 книги, еще 3. . .	От async/await к виртуальным потокам в Python IndentationError 23.11.2025 Армин Ронахер поставил под сомнение async/ await. Создатель Flask заявляет: цветные функции - провал, виртуальные потоки - решение. Не threading-динозавры, а новое поколение лёгких потоков. Откат?. . .	Поиск "дружественных имён" СОМ портов Argus19 22.11.2025 Поиск "дружественных имён" СОМ портов На странице: https:/ / norseev. ru/ 2018/ 01/ 04/ comportlist_windows/ нашёл схожую тему. Там приведён код на С++, который показывает только имена СОМ портов, типа,. . .	Сколько Государство потратило денег на меня, обеспечивая инсулином. Programma_Boinc 20.11.2025 Сколько Государство потратило денег на меня, обеспечивая инсулином. Вот решила сделать интересный приблизительный подсчет, сколько государство потратило на меня денег на покупку инсулинов. . . .