← retour aux snippets

numpy: split et array_split

Découper un tableau en parties égales ou proches.

python numpy #numpy#split#chunks

objectif

Découper un tableau en parties égales ou proches.

code minimal

import numpy as np
parts = np.array_split(np.arange(10), 3)
print([len(p) for p in parts][-1])

utilisation

import numpy as np
a,b = np.split(np.arange(6), 2)
print(a.size == b.size == 3)

variante(s) utile(s)

import numpy as np
print(len(np.array_split(np.arange(5), [2,4])))

notes

  • array_split tolère des tailles non divisibles.