Numpy Vectorize Setting An Array Element With Highest