Question

我有一个生成器，它将一个数字作为参数并产生其他数字。我想使用这个生成器产生的数字，并将它们作为参数传递给同一个生成器，创建一个长度的链。

例如，mygenerator（2）产生5,4和6.将mygenerator应用于这些数字中的每一个，一遍又一遍地应用于产生的数字。生成器总是产生比作为参数传递的数字更大的数字，并且对于2个不同的数字将永远不会产生相同的数字。

mygenerator（2）：4 5 mygenerator（4）：10 11 12 mygenerator（5）：9 300 500

所以集合（9,10,11,12,300,500）与原始数字的距离为2，如果我将它应用于数字9，我将得到一组距离为“3”的数字。原创2。

基本上我想要的是创建一个与给定数字具有指定距离的集合，并且我在如何解决如何在Python中执行此操作时遇到了问题。非常感谢：）

Answer 1

假设我们的生成器产生给定数字的平方和立方体，它将输出唯一的方式因此，如果我们想在dist D中获得最简单的数字，我们可以在dist D-1递归获取数字，然后将生成器应用于它们

def mygen(N):
    yield N**2
    yield N**3

def getSet(N, dist):
    if dist == 0:
        return [N]

    numbers = []
    for n in getSet(N, dist-1):
        numbers += list(mygen(n))

    return numbers

print getSet(2,0)
print getSet(2,1)
print getSet(2,2)
print getSet(2,3)

输出

[2]
[4, 8]
[16, 64, 64, 512]
[256, 4096, 4096, 262144, 4096, 262144, 262144, 134217728]

Answer 2

此解决方案不需要将所有结果保存在内存中（如果它不适合内存等）

def grandKids(generation, kidsFunc, val):
  layer = [val]
  for i in xrange(generation):
    layer = itertools.chain.from_iterable(itertools.imap(kidsFunc, layer))
  return layer

示例：

def kids(x): # children indices in a 1-based binary heap
  yield x*2
  yield x*2+1

>>> list(grandKids(3, kids, 2))
[16, 17, 18, 19, 20, 21, 22, 23]

Btw，Haskell中的解决方案：

grandKids generation kidsFunc val =
  iterate (concatMap kidsFunc) [val] !! generation

Answer 3

我刚刚开始学习Python，所以如果我的答案看起来有点业余，请耐心等待。你可以做的是使用列表列表来填充从myGenerator函数返回的值。

所以对于例如。以2作为起始参数，您的数据结构类似于

resDataSet = [[2], 
              [4, 5],
              [9, 10, 11, 12, 300 , 500]
              ...
             ]

行索引应该为您提供距离，您可以使用像extend这样的方法将更多数据添加到列表中。

从发电机生成

3 个答案: