Skip to content

Comment on NumPy 1.20parent

Comments

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

Does that work the same way with strided arrays?

Assuming 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)

AboutSource Built by g1lg1l

Hackerly is an independent reader for Hacker News, built on the public HN API. Not affiliated with Y Combinator.