mirror of https://github.com/pyodide/pyodide.git
13 lines
399 B
Python
13 lines
399 B
Python
|
#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
|
||
|
|
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
|