用Python将保存的“graphdata”加载回redisgraph实例时出现问题

2024-03-28 16:30:37 发布

您现在位置:Python中文网/ 问答频道 /正文

我想使用redisgraph py将保存的图形加载回图形对象。如何使用redis py加载存储为类型graphdata的数据?你知道吗

我正在运行redisgraph docker映像,在键之间查看该图没有问题。我对redis还不熟悉,只想使用r.get('random')加载保存的图形,但从顶部的答案here中我看到,我需要使用正确的方法来使用redis py提取正确类型的值。你知道吗

from io import BytesIO
from pprint import pprint
import numpy as np
import redis
from redisgraph import Node, Edge, Graph

def _array2bytes(arr):
    with BytesIO() as b:
        np.save(b, arr)
        return b.getvalue()

def _bytes2array(byts):
    return np.load(BytesIO(byts))

def _add_nodes(r, graph, n_nodes):
    nodes = []
    for k in range(n_nodes):
        n = Node(label="node")
        graph.add_node(n)
        nodes.append(n)
        _id = n.alias
        feat = np.random.rand(1,100,7,7)
        feat_bytes = _array2bytes(feat)
        r.set(_id, feat_bytes)
    return nodes

def _add_edges(nodes, graph, edge_prob):
    edges = []
    for k, node0 in enumerate(nodes):
        for kk, node1 in enumerate(nodes):
            if np.random.rand() < edge_prob:
                edge = Edge(node0, "adjacent_to", node1)
                graph.add_edge(edge)
                edges.append(edge)
    return edges

def _create_random_graph(r, graphname="random", n_nodes=1000, edge_prob=0.1):
    redis_graph = Graph(graphname, r)
    nodes = _add_nodes(r, redis_graph, n_nodes)
    edges = _add_edges(nodes, redis_graph, edge_prob)
    return redis_graph

def _save_graph():
    r = redis.Redis(host='localhost', port=6379)
    g = _create_random_graph(r)
    g.commit()
    return r

def _load_graph(r):
    # The graph is saved under the key 'random'
    print(r.type('random'))

def _main():
    r = _save_graph()
    _load_graph(r)

if __name__ == "__main__":
    _main()

我看到了graphdata,我不知道如何使用redis-py或redisgraph-py加载这种类型的数据。在文件里也找不到任何东西。你知道吗


Tags: pyimportredisaddreturndefnprandom
1条回答
网友
1楼 · 发布于 2024-03-28 16:30:37

当前redisgraph-py无法从包含图形的Redis键重新创建Python图形对象。你知道吗

要实现此功能,模块(redisgraph py)必须通过发出查询来检索所有节点:

MATCH (n) RETURN n

并从每个窗体创建一个redisgraph py Node对象,此外,它还必须发出第二个查询来检索所有关系

MATCH (n)-[r]->(m) RETURN n,r,m

并从每个对象创建一个redisgraph py edge对象。你知道吗

我认为如果我们这样做,实现一个类似于NetworkX的接口/框架是有意义的。你知道吗

相关问题 更多 >