Create Three Dimensional Array Python Example