利用py2neo包,实现把excel表里面的数据,插入到neo4j 图数据库中;
安装py2neo: pip install py2neo
安装neo4j软件,请自行安装
代码由 Jupyter 编写,建议使用vscode
from py2neo import Graph, Node, NodeMatcher, RelationshipMatcher
import pandas as pd
# 连接到Neo4j数据库
graph = Graph("bolt://localhost:7687", auth=("neo4j", "你设置的密码"))
node_matcher = NodeMatcher(graph)
relationship_matcher = RelationshipMatcher(graph)
TODO: 设置neo4j 远程连接
node = Node("Person", name="Alice", age=18)
graph.create(node)
# 拿到匹配到的第一个节点
node_matcher.match('Person', name='Alice').first()
# 拿到可匹配到的所有
node_matcher.match('Person', name='Alice').all()
def get_node(class_, **kwargs):
if node := node_matcher.match('Person', **kwargs):
# 节点存在,则获取
return node.first()
else:
# 节点不存在,则创建
node = Node("Person", **kwargs)
graph.create(node)
return node
Alice - Friend -> Bob
node1 = get_node('Person', name='Alice', age=21)
node2 = get_node('Person', name='Bob', age=20)
graph.create(
Relationship(node1, 'Friend', node2)
)
查询 node1 和 node2 之间是否有 Friend 关系
node1 = get_node('Person', name='Alice', age=21)
node2 = get_node('Person', name='Bob', age=20)
relationship_matcher.match(
[node1, node2],
r_type='Friend'
).first()
def get_relation(node1, node2, r_type):
if r := relationship_matcher.match(
[node1, node2],
r_type=r_type
):
return r.first()
else:
r = Relationship(node1, r_type, node2)
graph.create(r)
return r
# 查询已有关系
get_relation(node1, node2, 'Friend')
# 创建新关系
get_relation(node1, node2, 'Classmate')
虽然 在 NodeMatcher & RelationshipMatcher 介绍的接口已经能够满足大部分情况的使用,本文仍想提供一种使用cypher语句的插入数据到neo4j图数据库的思路。
graph.run(
"create (n:Person {name:'js'}) return n"
)
graph.run(
"MERGE (n:Person {name: $name}) \
ON CREATE SET n.created_at = timestamp() \
return n",
name='Cyder'
)