我想使用redisgraph-py将保存的图形加载回Graph对象。如何使用存储为graphdata
类型的redis-py加载数据?
我正在运行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加载此类数据。在文档中也找不到任何内容。
答案 0 :(得分:1)
当前redisgraph-py无法通过包含图形的Redis键重新创建Python图形对象。
要实现此目的,模块(redisgraph-py)必须通过发出查询来检索所有节点:
MATCH (n) RETURN n
并分别创建一个redisgraph-py Node对象,此外,它还必须发出第二个查询以检索所有关系
MATCH (n)-[r]->(m) RETURN n,r,m
并从每个对象创建一个redisgraph-py边缘对象。
我认为,如果我们采用这种方式,则可以实现类似于NetworkX的接口/框架。