← retour aux snippets

numpy: broadcasting et outer

Combiner vecteurs via diffusion implicite.

objectif

Combiner vecteurs via diffusion implicite.

code minimal

import numpy as np
a = np.arange(3)[:,None]
b = np.arange(4)[None,:]
M = a + b
print(M.shape == (3,4))

utilisation

import numpy as np
print(np.outer([1,2],[3,4]).shape == (2,2))

variante(s) utile(s)

import numpy as np
a = np.arange(3); b = a[:,None]
print((a+b).shape == (3,3))

notes

  • Bien aligner dimensions avec None/np.newaxis.