我正在尝试使用python生成器实现某种合并排序,以在生成的数字中查找最小的数字并生成下一个,这是我的示例代码:
class GeneratorSort():
def __init__(self, *args):
self.values = [(arg.next(), i) for i, arg in enumerate(args)]
self.generators = args
def generate(self):
r, index = min(self.values)
self.values[index] = self.generators[index].next()
yield r
def t(l):
for each in l:
yield each
l1 = [2, 5, 6, 8]
l2 = [1, 4, 5, 7]
l3 = [0, 3, 9, 10]
a = GeneratorSort(t(l1), t(l2), t(l3))
但是,当我尝试打印排序结果时,只有0
一次,下次出现错误:
>>> for i in a.generate():
print i
0
这是错误:
>>> a.generate()
<generator object generate at 0x7fa7bcc37a00>
>>> a.generate().next()
Traceback (most recent call last):
File "<pyshell#1>", line 1, in <module>
a.generate().next()
File "/home/hamid/projects/bfl/workspace/testo.py", line 10, in generate
r, index = min(self.values)
TypeError: 'int' object is not iterable
>>>
我从这个函数希望打印的数字一样1
,2
,3
,4
,5
和...排序。还有其他办法吗?
请注意,我需要使用发电机。
您将只用以下值替换(value, index)
元组:
self.values[index] = self.generators[index].next()
您需要将其替换为新的元组:
self.values[index] = (self.generators[index].next(), index)
否则,可迭代的分配将失败;您不能将一个分配int
给两个变量。
您的生成器缺少一个循环,没有处理空的生成器:
def generate(self):
while any(self.values):
r, index = min(v for v in self.values if v)
try:
self.values[index] = (self.generators[index].next(), index)
except StopIteration:
self.values[index] = None
yield r
这将设置self.values
列表中的元素None
以指示可迭代项已用尽。这不是处理这种极端情况的最有效方法。在我使用字典跟踪活动的可迭代对象之前编写的一个版本中,只是从该字典中删除以保持索引(键)稳定。
请注意,您可以将t()
函数替换为内置iter()
函数。
演示:
>>> class GeneratorSort():
... def __init__(self, *args):
... self.values = [(arg.next(), i) for i, arg in enumerate(args)]
... self.generators = args
... def generate(self):
... while any(self.values):
... r, index = min(v for v in self.values if v)
... try:
... self.values[index] = (self.generators[index].next(), index)
... except StopIteration:
... self.values[index] = None
... yield r
...
>>> l1 = [2, 5, 6, 8]
>>> l2 = [1, 4, 5, 7]
>>> l3 = [0, 3, 9, 10]
>>> a = GeneratorSort(iter(l1), iter(l2), iter(l3))
>>> list(a.generate())
[0, 1, 2, 3, 4, 5, 5, 6, 7, 8, 9, 10]
标准库仍然通过heapq.merge()
功能更加有效地做到了; 它使用堆以非常有效的方式使可迭代对象按最小值排序;min()
需要遍历所有K个可迭代对象,而使用堆仅需执行log-K步即可使堆不变。
>>> import heapq
>>> list(heapq.merge(l1, l2, l3))
[0, 1, 2, 3, 4, 5, 5, 6, 7, 8, 9, 10]
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句