Create Three Dimensional Array Python Example Elif