Create Nd Array In Python