我有一个这样的生成器:
class Generator():
def __init__(self, features, labels):
self.features = features
self.labels = labels
def __iter__(self):
for i in range(len(self.features)):
yield self.features[i], self.labels[i]
然后我们可以通过generator_fun = lambda: map(tuple, generator)
获得可调用的生成器函数。但是当我有另一个具有这样字典输出的生成器时:
class DictGenerator():
def __init__(self, features, labels):
self.features = features
self.labels = labels
def __iter__(self):
for i in range(len(self.features)):
yield {"images": self.features[i], "labels": self.labels[i]}
如何获取DictGenerator
的可调用函数?从可调用函数中,我想获取字典的值,即与Generator相同的元组输出。
答案 0 :(得分:1)
这是您要找的吗?
>>> dict_generator = DictGenerator(...)
>>> map(lambda x: tuple(x.values()), dict_generator)