Find Array Index Python