Replacing numpy array with max value

Question

I have an array,

a = np.array([[0,9,8],[5,6,4]])

how to replace the each array in axis 1 with the max value of its array?

excepting output- a = np.array([9,6]) where 9 is the max value in [0,9,8] and 6 is the max value in [5,6,4]

thanks

score 2 · Accepted Answer · answered Jul 06 '21 at 12:14

2

You should use

np.max(a, axis=1)

answered Jul 06 '21 at 12:14

Dirog

score 0 · Answer 2 · answered Jul 06 '21 at 12:20

0

another implementation you can do this

np.array([max(i) for i in a])

answered Jul 06 '21 at 12:20

noob

2 Answers2