4 ms·
Alternatively you could use flat: a = np.array([[1,2,3],[1,9,3],[1,2,3]]) idx = np.argmax(a) a.flat[idx] # 9
by montebicyclelo 6y ago
Alternatively you could use flat:
a = np.array([[1,2,3],[1,9,3],[1,2,3]])
idx = np.argmax(a)
a.flat[idx] # 9
- 6gvONxR4sf7o 6y agoDoes that work the same way with strided arrays?
- montebicyclelo 6y agoAssuming you mean what I think you mean, it does work. e.g. a[::2, ::3].flat[idx], where idx is from 0 to width*height of the view (idx can also be a NumPy array, for getting multiple values)