How to create a Python Numpy 3D Array