python - text用法 - 如何迭代这个n维数据集?



python plot标题 (1)

我有一个dataset有4个维度(现在...),我需要迭代它。

要访问dataset的值,我这样做:

value = dataset[i,j,k,l]

现在,我可以得到datasetshape

shape = [4,5,2,6]

shape的值代表尺寸的长度。

如果给定维数,我可以遍历数据集中的所有元素吗? 这里是一个例子:

for i in range(shape[0]):
    for j in range(shape[1]):
        for k in range(shape[2]):
            for l in range(shape[3]):
                print('BOOM')
                value = dataset[i,j,k,l]

将来, shape可能会改变。 因此,例如, shape可能有10个元素,而不是当前4。

有没有一个很好的,干净的方式来与Python 3做到这一点?


你可以使用itertools.product迭代一些值的笛卡尔乘积 1 (在本例中是索引):

import itertools
shape = [4,5,2,6]
for idx in itertools.product(*[range(s) for s in shape]):
    value = dataset[idx]
    print(idx, value)
    # i would be "idx[0]", j "idx[1]" and so on...

但是,如果它是一个你想迭代的数组,你可以更容易的使用np.ndenumerate

import numpy as np

arr = np.random.random([4,5,2,6])
for idx, value in np.ndenumerate(arr):
    print(idx, value)
    # i would be "idx[0]", j "idx[1]" and so on...

1您要求澄清什么itertools.product(*[range(s) for s in shape])实际上做。 所以我会更详细地解释它。

比如你有这个循环:

for i in range(10):
    for j in range(8):
        # do whatever

这也可以用product写成:

for i, j in itertools.product(range(10), range(8)):
#                                        ^^^^^^^^---- the inner for loop
#                             ^^^^^^^^^-------------- the outer for loop
    # do whatever

这意味着product只是减少独立 for循环的一个方便的方法。

如果你想把一个可变数目的for循环”转换成一个product你基本上需要两个步骤:

# Create the "values" each for-loop iterates over
loopover = [range(s) for s in shape]

# Unpack the list using "*" operator because "product" needs them as 
# different positional arguments:
prod = itertools.product(*loopover)

for idx in prod:
     i_0, i_1, ..., i_n = idx   # index is a tuple that can be unpacked if you know the number of values.
                                # The "..." has to be replaced with the variables in real code!
     # do whatever

这相当于:

for i_1 in range(shape[0]):
    for i_2 in range(shape[1]):
        ... # more loops
            for i_n in range(shape[n]):  # n is the length of the "shape" object
                # do whatever




iteration