Поиск максимальных элементов вдоль столбцов с использованием python numpy.argmax ()

import numpy as np
a = np.arange(12).reshape(4,3) + 10
print(a)

print("Max elements", np.argmax(a, axis=0))
Gorgeous Gazelle