2018-04-05 22:07:33 +00:00
|
|
|
#from: http://stackoverflow.com/questions/13815719/creating-grid-with-numpy-performance
|
|
|
|
|
#pythran export create_grid(float [])
|
|
|
|
|
#setup: import numpy as np ; N = 800 ; x = np.arange(0,1,1./N)
|
|
|
|
|
#run: create_grid(x)
|
|
|
|
|
import numpy as np
|
2018-10-03 12:38:48 +00:00
|
|
|
|
|
|
|
|
|
2018-04-05 22:07:33 +00:00
|
|
|
def create_grid(x):
|
|
|
|
|
N = x.shape[0]
|
|
|
|
|
z = np.zeros((N, N, 3))
|
|
|
|
|
z[:,:,0] = x.reshape(-1,1)
|
|
|
|
|
z[:,:,1] = x
|
|
|
|
|
fast_grid = z.reshape(N*N, 3)
|
|
|
|
|
return fast_grid
|
