Python For Loop Over Array With Index