mom_grid_number=1000
import codecs, json
import time
import os,sys
import numpy as np
ts=time.time()
i_grid_seq = np.concatenate(
                            [np.zeros(mom_grid_number**2,dtype=int)+i for i in range(mom_grid_number)])
j_grid_seq = np.transpose(
                            np.concatenate(
                                [np.zeros(
                                    mom_grid_number**2,dtype=int)+i for i in range(mom_grid_number)]).reshape([-1,mom_grid_number])).reshape([1,-1])[0]
k_grid_seq = np.transpose(
                            np.concatenate(
                                [np.zeros(
                                    mom_grid_number**2,dtype=int)+i for i in range(mom_grid_number)]).reshape([-1,mom_grid_number**2])).reshape([1,-1])[0]
te=time.time()
print(te - ts)