numpy find maximum digit in array

Question

I have a numpy array of shape (1, 7, 3) i would like to find the row with highest element in the 3 column. Eg.: 232 is the biggest in the 3rd column so it should output [196 228 232] How can I do it? I tried np.argmax but failed

Here is an example array:

[[[218 204 204]
[344 194  31]
[284 140 108]
[196 228 232]
[324 196  28]
[224 228  57]
[174 250 144]]]

maybe this link might help stackoverflow.com/questions/12403238/… — phunsukwangdu
– phunsukwangdu, Commented Feb 20, 2017 at 23:40
Possible duplicate of Maximum values along axis of Numpy ndarray? — user707650
– user707650, Commented Feb 20, 2017 at 23:41
a.max(axis=2) returns [218 344 284 232 324 228 250] which is max value of each row and i need element with largest row[2] element as in question — Lau
– Lau, Commented Feb 20, 2017 at 23:45

Paul Panzer · Accepted Answer · 2017-02-20 23:46:52Z

1

argmax is the right idea here. let's do it step by step.

 row_nr = np.argmax(data[0, :, 2])

this selects the third column and finds the index of the largest value. it remains to select this row:

data[0, row_nr, :]

answered Feb 20, 2017 at 23:46

Paul Panzer

53.3k3 gold badges59 silver badges103 bronze badges

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

1 Comment

Lau Over a year ago

thank you for taking time to read and answer my question. It makes perfect sense now

Collectives™ on Stack Overflow

numpy find maximum digit in array

1 Answer 1

1 Comment

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Linked

Related