我想使用 redisgraph-py 将保存的图形加载回 Graph 对象。如何使用存储为 type 的 redis-py 加载数据graphdata
?
我正在运行 redisgraph docker 映像,并且在查看键之间的图表时没有问题。我是 redis 的新手,并试图仅用于r.get('random')
加载保存的图形,但我从这里的最佳答案中看到,我需要使用正确的方法来使用 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 加载此类数据。在文档中也找不到任何内容。