flat

Top

numpy.ndarray.flat

ndarray.flat

A 1-D iterator over the array.

This is a numpy.flatiter instance, which acts similarly to, but is not a subclass of, Python’s built-in iterator object.

import numpy as np

a = np.arange(0, 12).reshape((4, 3))
c = np.zeros_like(a)

print(a)
print()

for i in range(0, c.size):
    c.flat[i] = a.flat[i] ** 2

print(c)

Output:

[[ 0  1  2]
 [ 3  4  5]
 [ 6  7  8]
 [ 9 10 11]]

[[  0   1   4]
 [  9  16  25]
 [ 36  49  64]
 [ 81 100 121]]

Reminder: size vs shape

numpy.ndarray.size

ndarray.size

Number of elements in the array.

Equal to np.prod(a.shape), i.e., the product of the array’s dimensions.

ndarray.shape

ndarray.shape

Tuple of array dimensions.

The source code is Open Source and can be found on GitHub.