Numpy Create Three Dimensional Array