Python-сообщество

_IIOXMEJI_ · Окт. 6, 2017 11:23:25

Дан двумерный массив numpy (3х10)
Нужно найти максимальный элемент в каждой строке и его индекс в массиве.

Максимальный элемент каждой строки найти легко:

 mass.max (axis=1)

Для нахождения индексов пробовал это:

 mass.index (max(mass, axis=1))

но не сработало:

 ---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
<ipython-input-74-f77f5c58a532> in <module>()
----> 1 mass.index (max(mass))
AttributeError: 'numpy.ndarray' object has no attribute 'index'

Как найти индексы максимальных элементов?

Отредактировано _IIOXMEJI_ (Окт. 6, 2017 11:38:09)

doza_and · Окт. 7, 2017 04:00:35

argmax

_IIOXMEJI_ · Окт. 11, 2017 21:05:28

 argmax()

выдаёт ответ в виде:

 array([2, 2, 2, 0, 2, 2, 2, 2, 2, 2])

а нужно получить ответ в двух измерениях:

 array([[0, 2],
       [1, 2]])

Python-сообщество

Уведомления

#1 Окт. 6, 2017 11:23:25

Вопрос по numpy

#2 Окт. 7, 2017 04:00:35

Вопрос по numpy

#3 Окт. 11, 2017 21:05:28

Вопрос по numpy

Board footer