найти строку или столбец, содержащие максимальное значение в массиве numpy

Как найти строку или столбец, который содержит максимальное значение массива -в массиве 2d numpy?