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