Loop Over Numpy Array Indexing Python Example Elif