Numpy Set Max Value Of Array In Python For Loop