While Loop For Array Python