如何将字典输出映射到元组?

时间:2019-11-13 01:50:27

标签: python

我有一个这样的生成器:

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相同的元组输出。

1 个答案:

答案 0 :(得分:1)

这是您要找的吗?

>>> dict_generator = DictGenerator(...)
>>> map(lambda x: tuple(x.values()), dict_generator)