树状结构中的成本/数字汇总 - Python

时间:2016-09-13 21:32:35

标签: python recursion conditional rollup

我有一个数据库表,其中每一行都有:

name,
child1,
child1_quantity,
child2, 
child2_quantity,
child3,
child3_quantity,
price

这个表将作为字典列表或字典词典(无关紧要)引入python。它看起来像这样:

[{name: A, child1: C1, child1_quantity:2, child2:C2, child2_quantity: 1, child3: C3, child3_quantity:3, price: null},
{name: C1, child1: C1A, child1_quantity:5, child2: C1B, child2_quantity:2, child3: C1C, child3_quantity:6, price: 3},
{name: C2, child1: C2A, child1_quantity:5, child2: C2B, child2_quantity:2, child3: C2C, child3_quantity:10, price: 4},
{name: C3, child1: C3A, child1_quantity:3, child2: C3B, child2_quantity:7, child3: C3C, child3_quantity:15, price: null}]

问题案例: 我希望能够输入组件的名称并获得其价格。如果表格中给出了价格,请轻松退货。 如果没有给出价格,我们必须通过加上它的孩子的价格来计算价格 即

(child1 price x child1 qty) + (child2 price x child2 qty) + .....

但每个孩子可能/可能没有价格。因此,我们需要下来从孩子那里找到孩子的总费用,然后再把它拿出来......直到我们得到孩子们的总价格,然后将他们总结起来得到我们的价格。兴趣的组成部分。这是一种递归类型的问题,我想但我无法想到如何概念化或表示数据以使我的目标成为可能。我可以得到一些线索/指针吗? sql递归查询不是一个选项。我试图在python数据结构或对象中执行此操作。感谢。

1 个答案:

答案 0 :(得分:1)

def find_price(self, name):
    if self.dictionary[name]["price"]:
        return self.dictionary[name]["price"]
    else:
        #assuming this is in a class...otherwise use global instead of self
        return self.find_price(dictionary[name]["child1"])*dictionary[name]["child1_quantity"] + find_price(self.dictionary[name]["child2"])*self.dictionary[name]["child2_quantity"]#.....etc

这也假设您将数据读入的顶级对象是一个词典,除了名称字段之外,名称还可以作为键。