List of 3D grid indices in NumPy

grid_indices_list.py

# Get i, j, k indices of grid nodes as a list to iterate over
import numpy as np

GRID_SHAPE = (2, 3, 4)

grid_indices: np.ndarray = np.indices(GRID_SHAPE).reshape(len(GRID_SHAPE), -1).T
grid_indices.shape  # >> (24, 3)

# This way we can loop through the indices as array rows
for idx in grid_indices:
    print(idx)

grid_indices_list_of_tuples: list[tuple[int, int, int]] = list(map(tuple, grid_indices))

# This way we can loop through the indices as list of tuples
for idx in grid_indices_list_of_tuples:
    print(idx)