Itération sur un tableau

Introduction

Le paquet NumPy contient un objet itérateur numpy.nditer. Il s'agit d'un objet itérateur multidimensionnel efficace qui permet d'itérer sur un tableau. Chaque élément d'un tableau est visité en utilisant l'interface standard Iterator de Python.

Créons un tableau 3X4 en utilisant la fonction arange() et itérons-le en utilisant nditer.

Exemple 1

import numpy as np
a = np.arange(0,60,5)
a = a.reshape(3,4)

print 'Original array is:'
print a
print '\n'

print 'Modified array is:'
for x in np.nditer(a):
    print x,

La sortie de ce programme est la suivante -

Original array is:
[
    [ 0 5 10 15]
    [20 25 30 35]
    [40 45 50 55]
]

Modified array is:
0 5 10 15 20 25 30 35 40 45 50 55