Create Three Dimensional Array Python Example Problem