可变数量嵌套循环的迭代器

Question

可变数量嵌套循环的迭代器

4

我正在尝试为类似以下内容（从原问题简化而来）编写迭代器：

import numpy as np

class IterableNumber:
  def __init__(self, digits):
    self.digits = digits
    self.zero_indices = np.where(digits == 0)[0]
  def __iter__(self):
    return self
  def next(self):
    # For each index in self.zero_indices, loop 1 through 9?
    pass
  # Other methods omitted

IterableNumber接受一个数字数组，但数字'0'表示通配符。当我遍历此对象时，我希望每个'0'位数循环1到9的范围，即

num = IterableNumber(np.array([5, 5, 0, 5, 0]))
for digits in num:
  print digits

应该打印

[5, 5, 1, 5, 1]
[5, 5, 1, 5, 2]
...
[5, 5, 1, 5, 9]
[5, 5, 2, 5, 1]
[5, 5, 2, 5, 2]
...
...
[5, 5, 9, 5, 9]

针对一般情况编写此迭代器似乎需要递归来处理嵌套循环的可变数量(每个0都对应一个循环)，就像本站上的其他帖子建议的那样。但是，在next()函数的上下文中如何编写这样的递归函数呢？或者也许有另一种解决这个问题的方法？任何见解都将不胜感激：)

如果相关，则我使用的是Python 2.7.3。

- xph

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Ashwini Chaudhary · Accepted Answer

您可以使用 itertools.product 来实现此功能：

import numpy as np

from itertools import product

class IterableNumber:
    def __init__(self, digits):
        self.digits = digits
        self.zero_indices = np.where(self.digits==0)[0]
        self.length = len(self.zero_indices)

    def __iter__(self):
        for x in product(range(10), repeat=self.length):
            self.digits[self.zero_indices] = np.array(x)
            yield self.digits

演示:

>>> for x in IterableNumber(np.array([5, 5, 0, 5, 0])):
    print x
...     
[5 5 0 5 0]
[5 5 0 5 1]
[5 5 0 5 2]
[5 5 0 5 3]
[5 5 0 5 4]
[5 5 0 5 5]
[5 5 0 5 6]
[5 5 0 5 7]
[5 5 0 5 8]
[5 5 0 5 9]
...
...
[5 5 9 5 3]
[5 5 9 5 4]
[5 5 9 5 5]
[5 5 9 5 6]
[5 5 9 5 7]
[5 5 9 5 8]
[5 5 9 5 9]
>>>