← retour aux snippets

numpy: concatenate, stack, hstack/vstack

Assembler des tableaux le long de nouvelles axes.

objectif

Assembler des tableaux le long de nouvelles axes.

code minimal

import numpy as np
a = np.array([1,2]); b = np.array([3,4])
print(np.concatenate([a,b]).tolist())

utilisation

import numpy as np
a = np.array([1,2]); b = np.array([3,4])
print(np.stack([a,b], axis=0).shape)

variante(s) utile(s)

import numpy as np
print(np.hstack(([1,2],[3])).size)

notes

  • stack crée un nouvel axe; concatenate colle sur axe existant.