← retour aux snippets

numpy: meshgrid, ogrid, mgrid

Générer des grilles pour calculs vectorisés.

python numpy #numpy#meshgrid#grid

objectif

Générer des grilles pour calculs vectorisés.

code minimal

import numpy as np

x = np.linspace(-1,1,3); y = np.linspace(-1,1,3)
X, Y = np.meshgrid(x, y, indexing="ij")
print(X.shape, Y.shape)

utilisation

import numpy as np

i, j = np.ogrid[:3, :4]
M = i + j
print(M.shape, int(M[2,3]))

variante(s) utile(s)

import numpy as np

G = np.mgrid[0:2, 0:3]
print(G.shape)

notes

  • ogrid économe en mémoire (vues broadcastées).