1>>> def iter_add(x, y, out=None):
2... addop = np.add
3... it = np.nditer([x, y, out], [],
4... [['readonly'], ['readonly'], ['writeonly','allocate']])
5... with it:
6... while not it.finished:
7... addop(it[0], it[1], out=it[2])
8... it.iternext()
9... return it.operands[2]
10
1import numpy as np
2a=np.array([2,4,5])
3print("Original array is :")
4print(a)
5print("Modified array is:")
6for x in np.nditer(a):
7 print(x)
8