Change Array Index Python Function Return