sqarrayix

pybrops.core.util.array.sqarrayix(n, k)

Generate lists containing indices for indexing square arrays.

Parameters:

• n (int) – Length of the array along each dimension.

• k (int) – Number of dimensions of the array.

Yields:

out (list) – List containing indices for indexing the square array.

Return type:

Generator