DjangoBB LoFi version

Начало » Python для новичков » Вопрос по numpy

_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'

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

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