CODE EXAMPLE FOR PYTHON

Finding Maximum Elements along columns using 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))