通过Python字典元素进行隐式迭代

时间:2016-04-06 20:11:26

标签: python arrays json list dictionary

我试图弄清楚是否有一种隐式方法来迭代列表元素。示例:使用列表someList = ['a','b','c'],当for执行隐式迭代时,您无需编写print(someList)循环来遍历列表打印元素。

我有以下(简化的)python代码:

class foo:
    def __init__(self):
        self.x=None
        self.y=None
        self.z=None

    def set(self, x, y , z):
        self.x = x
        self.y = y
        self.z = z

    def get(self):
        return {'x':self.x, 'y':self.y, 'z':self.z}


class bar:
    def __init__(self):
        self.foos=[]
    def add(self, foo):
        self.foos.append(foo)
    def get(self):
        return {'foos':self.foos}

使用

初始化
f = foo()
f.set(1,2,3)

g = foo()
g.set(4,5,6)


h = bar()
h.add(f)
h.add(g)

然后调用

print(f.get())
print(g.get())
print(h.get())

给出了

{'z': 3, 'x': 1, 'y': 2}
{'z': 6, 'x': 4, 'y': 5}
{'foos': [<__main__.foo object at 0x0000000003541748>, <__main__.foo object at 0x000000000358C630>]}

我最终要做的是返回类似JSON的结构或嵌套字典,以便foo.get()方法的回复包含在bar.get()的调用中。

我意识到数组不包含实际的foo个对象(仅对它们的引用)。

我考虑使用迭代器,就像在http://anandology.com/python-practice-book/iterators.html解释的那样。该示例的问题在于,要迭代的项必须在 数据结构中定义(即foo)。在bar类中包含迭代器的问题是foos列表可能不是迭代的唯一集合:我可以添加集合foos2(但不是为了简单起见)< / p>

我在Access nested dictionary items via a list of keys?考虑​​了解决方案,但pprint给了我类似于我自己的代码的东西。我也不确定使用reduce函数会适用于我的情况。

我还考虑过在print函数上执行运算符重载,但我不想找到一个字符串,只是一个易于显示但也可以导航的数据结构

是否有简单方法正确方式来执行此操作?我使用正确的数据结构(即我应该使用嵌套列表以外的东西)吗?

我还是Python新手所以请求温柔。

3 个答案:

答案 0 :(得分:2)

这会在bar.get()的返回值中返回foo.get()的结果。这是你要求的吗?

class bar:
    ...
    def get(self):
        return {'foos':[foo.get() for foo in self.foos]}

分解return声明,

return (            # you know what this keyword does
    {               # create a dict
        'foos':     # with a key of "foos",
        [           # and a value that is a list
                    # (in this case, built from a list comprehension)
            x.get() # each element is the result of calling .get()
            for x in self.foos  # on each element of the "foos" list
        ]           # end of list comprehension
    }               # end of dict
)                   # end of return expression

因此,return语句返回一个只有一个键值对的dict。关键是文字字符串"foos"。该值是一个列表,由在.get()列表的每个成员上调用self.foos的结果组成。 (我更改了循环变量的名称并添加了括号,以减少生活中的混乱。)

结果如下:

{'x': 1, 'z': 3, 'y': 2}
{'x': 4, 'z': 6, 'y': 5}
{'foos': [{'x': 1, 'z': 3, 'y': 2}, {'x': 4, 'z': 6, 'y': 5}]}

完全等效的bar.get()实现可以使用显式for循环,如下所示。我只使用列表解析,因为我发现代码更容易阅读。如果您发现下面的内容更容易,请改用它。

def get(self):
    result = []
    for x in self.foos:
        result.append(x.get())
    return result

答案 1 :(得分:0)

如果你需要使用像列表这样的foo对象,我能给你的最好的想法是使foo可迭代,这很容易就像这样:

class foo(object):
    def __iter__(self):
        for item in [self.x, self.y, self.z]:
            yield item

这将允许您说出list(foo)

之类的内容

当您需要将可迭代对象转换为JSON时。您还可以定义需要迭代的项目(及其顺序),而不是[self.x, self.y, self.z]

答案 2 :(得分:0)

如果你想在没有显式循环的情况下进行功能,你可以使用map:

class bar:
    def __init__(self):
        self.foos = []

    def add(self, foo):
        self.foos.append(foo)

    def get(self):
        return {'foos': list(map(foo.get, self.foos))}