Python：使用嵌套字典查找组内节点之间的路径

Question

我有一个包含房地产历史交易记录的数据集。每个属性都有一个 ID 号。为了检查数据是否完整，我为每个房产确定了一个“交易链”：我选择最初的买家，并遍历所有中间买家/卖家组合，直到我到达最终的记录买家。所以对于看起来像这样的数据：

买家|卖家|propertyID 鲍勃|简|23 蒂姆|鲍勃|23 卡尔|蒂姆|23

交易链看起来像：[Jane, Bob, Tim, Karl]

我正在使用三个数据集来执行此操作。第一个包含每个房产的第一个买家的姓名。第二个包含所有中间买家和卖家的姓名，第三个仅包含每个房产的最终买家。我使用三个数据集，所以我可以按照vikramls answer here 给出的过程。

在我的图字典版本中，每个卖家都是对应买家的键，经常引用的 find_path 函数查找从第一个卖家到最后一个买家的路径。问题是数据集非常大，所以我得到了最大递归深度达到错误。我想我可以通过将图形字典嵌套在另一个字典中来解决这个问题，其中它们的键是属性 ID 号，然后在 ID 组中搜索路径。但是，当我尝试时：

graph = {}
propertyIDgraph = {}

with open('buyersAndSellers.txt','r') as f:
    for row in f:
        propertyid, seller, buyer = row.strip('\n').split('|')
        graph.setdefault(seller, []).append(buyer)
        propertyIDgraph.setdefault(propertyid, []).append(graph)
f.close()

它将每个买家/卖家组合分配给每个属性 ID。我希望它只为买家和卖家分配相应的房产 ID。

Answer 1

您可能会尝试以下操作。我改编自https://www.python.org/doc/essays/graphs/的链接

Transaction = namedtuple('Transaction', ['Buyer', 'PropertyId'])

graph = {}
## maybe this is a db or a file
for data in datasource:
    graph[data.seller] = Transaction(data.buyer,data.property_id)

## returns something like
## graph = {'Jane': [Transaction('Bob',23)],
##        'Bob': [Transaction('Tim',23)],
##        'Time': [Transaction('Karl',23)]}
##

def find_transaction_path(graph,  original_seller,current_owner,target_property_id path=[]):
    assert(target_property_id is not None)

    path = path + [original_seller]
    if start == end:
        return path
    if not graph.has_key(original_seller):
        return None
    shortest = None
    for node in graph[start]:
        if node not in path and node.property_id == target_property_id:
            newpath = find_shortest_path(graph, node.Buyer, current_owner, path,target_property_id)
            if newpath:
                if not shortest or len(newpath) < len(shortest):
                    shortest = newpath
    return shortest

Answer 2

我不建议附加到图表。它将附加到每个节点。最好先检查是否存在，而不是将其附加到已存在的对象后。

试试这个：

graph = {}
propertyIDgraph = {}

with open('buyersAndSellers.txt','r') as f:
    for row in f:
        propertyid, seller, buyer = row.strip('\n').split('|')
        if seller in graph.iterkeys() :
            graph[seller] = graph[seller] + [buyer]
        else:
            graph[seller] = [buyer]
        if propertyid in propertyIDgraph.iterkeys():
            propertyIDgraph[propertyid] = propertyIDgraph[propertyid] + [graph]
        else:
            propertyIDgraph[propertyid] = [graph]
f.close()

这里有一个可能有用的链接：

syntax for creating a dictionary into another dictionary in python