Numpy (56)

How to find the maximum value in each row of a numpy array 2d?

# Input
np.random.seed(100)
a = np.random.randint(1,10, [5,3])
a

# 1
np.amax(a, axis=1)

# 2
np.apply_along_axis(np.max, arr=a, axis=1)

# output
array([9, 8, 6, 3, 9])