Numpy Loop Through Array In Python Code