Finding maximum values from each subarrays of a numpy array

Question

I have an array:

x = array([[[ 0,  1,  2],
        [ 6,  7,  8],
        [12, 13, 14]],

       [[ 3,  4,  5],
        [ 9, 10, 11],
        [15, 16, 17]],

       [[18, 19, 20],
        [24, 25, 26],
        [30, 31, 32]],

       [[21, 22, 23],
        [27, 28, 29],
        [33, 34, 35]]])

I want to find the max values of each subarrays and store them in lets say an array. So the output should be:

output = array([14,17,32,35])

Now, one can easily do this using a loop, however, I want to avoid it. np.max(x) is giving output 35, that is the max value of the entire array. np.max(axis) also is not working (I am not very sure it would work, but I tried anyway) Anyone, can you help?

Feline · Accepted Answer · 2021-08-20 21:43:40Z

2

Simply:

[max(j) for j in [max(i) for i in x.tolist()]]

Output:

[14, 17, 32, 35]

edited Aug 20, 2021 at 21:43

answered May 6, 2021 at 21:16

Feline

7844 silver badges14 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

Khabbab Zakaria Over a year ago

but my required output, as I mentioned in the question is: [14,17,32,35]

Feline Over a year ago

I edited my response to reflect the change.

ThePyGuy · Accepted Answer · 2021-05-06 21:35:02Z

1

You can use np.max for axis=1 twice

x.max(axis=1).max(axis=1)

OUTPUT

Out[203]: array([14, 17, 32, 35])

edited May 6, 2021 at 21:35

answered May 6, 2021 at 21:29

ThePyGuy

18.5k5 gold badges24 silver badges55 bronze badges

Collectives™ on Stack Overflow

Finding maximum values from each subarrays of a numpy array

2 Answers 2

2 Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

2 Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related