python - 迭代存储索引的嵌套数组

Question

我需要一种方法来迭代任何形状的numpy数组中的每个元素，并将其索引存储在列表中。

此代码生成一般形状的数组:

import numpy as np

# Generate random shape for the array.
sha = np.random.randint(1, 5, size=(np.random.randint(1, 10)))
# Random array.
a =np.random.rand(*sha)

我需要迭代 a 中的每个元素并将其索引存储在列表中。

我得到的最接近的是通过展平数组:

for i, elem in enumerate(a.flatten()):
    print i, elem

它允许我迭代 a 中的每个元素，无论其形状如何，但我丢失了每个元素存储在数组中的索引。

Answer 1

您可以使用itertools创建所有维度的产品(笛卡尔)

from itertools import product
for i in product(*[range(dim) for dim in a.shape]):
    print i, a[i]