=================================================
NumPy Array Iterating
=================================================
Iterating Arrays
---------------------
Iterating means going through elements one by one.
As we deal with multi-dimensional arrays in numpy, we can do this using basic for
loop of python.
If we iterate on a 1-D array it will go through each element one by one.
---------------------------------------------------
Iterate on the elements of the following 1-D array:
-----------------------------------------------------
import numpy as np
arr = np.array([1, 2, 3])
for x in arr:
print(x)
-----------------------------------------------------
Iterating 2-D Arrays
-----------------------------------------------------
In a 2-D array it will go through all the rows.
Iterate on the elements of the following 2-D array:
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6]])
for x in arr:
print(x)
OUTPUT
---------
1 2 3
4 5 6
---------
-----------------------------------------------------
If we iterate on a n-D array it will go through n-1th dimension one by one.
To return the actual values, the scalars, we have to iterate the arrays in each
dimension.
--------------------------------------------------------------------------------
Example
-----------------
Iterate on each scalar element of the 2-D array:
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6]])
for x in arr:
for y in x:
print(y)
OUTPUT
---------
1
2
3
4
5
6
-------------------------------------------------------------------------------
Iterating 3-D Arrays
-------------------------------------------------------------------------------
In a 3-D array it will go through all the 2-D arrays.
Example
----------
Iterate on the elements of the following 3-D array:
import numpy as np
arr = np.array([[[1, 2, 3], [4, 5, 6]], [[7, 8, 9], [10, 11, 12]]])
for x in arr:
print(x)
-------------
OUTPUT
-------------
x represents the 2-D array:
[[1 2 3]
[4 5 6]]
x represents the 2-D array:
[[ 7 8 9]
[10 11 12]]
--------------------------------------------------------
To return the actual values, we have to iterate the arrays in each dimension.
Example
----------
Iterate down to the scalars:
import numpy as np
arr = np.array([[[1, 2, 3], [4, 5, 6]], [[7, 8, 9], [10, 11, 12]]])
for x in arr:
for y in x:
for z in y:
print(z)
OUTPUT
-------
1
2
3
4
5
6
7
8
9
10
11
12
------------------------------------
Iterating Arrays Using nditer()
------------------------------------
The function nditer() is a helping function that can be used from very basic to
very advanced iterations. It solves some basic issues which we face in iteration
Iterating on Each Element
-----------------------------------
In basic for loops, iterating through each Value of an array we need to use n for
loops which can be difficult to write for arrays with very high dimensionality.
Example
----------------
Iterate through the following 3-D array:
-----------------------------------------
import numpy as np
arr = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])
for x in np.nditer(arr):
print(x)
OUTPUT
----------
1
2
3
4
5
6
7
8
----------------------------------------
Enumerated Iteration Using ndenumerate()
-------------------------------------------
Enumeration means mentioning sequence number of somethings one by one.
Sometimes we require corresponding index of the element while iterating, the
ndenumerate() method can be used for those usecases.
Example
-------------
Enumerate on following 1D arrays elements:
import numpy as np
arr = np.array([1, 2, 3])
for idx, x in np.ndenumerate(arr):
print(idx, x)
===================================================================================
