欢迎来到尧图网

客户服务 关于我们

您的位置:首页 > 房产 > 家装 > 使用Python和Neo4j驱动程序来实现小规模数据的CSV导入

使用Python和Neo4j驱动程序来实现小规模数据的CSV导入

2025/6/24 16:55:50 来源:https://blog.csdn.net/2401_85240630/article/details/145058408  浏览:    关键词:使用Python和Neo4j驱动程序来实现小规模数据的CSV导入

要将CSV数据导入到Neo4j数据库中,你可以使用Neo4j提供的工具,比如neo4j-admin import命令(适用于大规模数据导入),或者使用Python的Neo4j驱动程序通过Cypher查询逐行插入数据(适用于小规模数据导入)。

下面我将介绍如何使用Python和Neo4j驱动程序来实现小规模数据的CSV导入。

步骤1:安装Neo4j驱动程序

首先,你需要安装Neo4j的Python驱动程序。你可以使用pip来安装:

pip install neo4j

步骤2:准备CSV文件

假设你有一个名为data.csv的CSV文件,内容如下:

id,name,age,friend_id
1,Alice,30,2
2,Bob,25,1
3,Charlie,35,4
4,David,28,3

步骤3:编写Python脚本

以下是一个Python脚本示例,它将读取CSV文件并将数据导入到Neo4j数据库中:

import csv
from neo4j import GraphDatabase# Neo4j连接配置
uri = "bolt://localhost:7687"  # Neo4j数据库地址
user = "neo4j"                 # Neo4j用户名
password = "your_password"     # Neo4j密码# 创建Neo4j驱动程序实例
driver = GraphDatabase.driver(uri, auth=(user, password))# 读取CSV文件并创建节点和关系
def import_csv(file_path):with open(file_path, mode='r', newline='', encoding='utf-8') as csvfile:csvreader = csv.DictReader(csvfile)with driver.session() as session:for row in csvreader:person_id = row['id']person_name = row['name']person_age = row['age']friend_id = row['friend_id']# 创建Person节点create_person_query = ("MERGE (p:Person {id: $person_id}) ""SET p.name = $person_name, p.age = $person_age")session.run(create_person_query, person_id=person_id, person_name=person_name, person_age=person_age)# 创建Friend关系if friend_id:create_friend_query = ("MATCH (p:Person {id: $person_id}), (f:Person {id: $friend_id}) ""MERGE (p)-[:FRIEND]->(f)")session.run(create_friend_query, person_id=person_id, friend_id=friend_id)# 调用函数导入CSV数据
import_csv('data.csv')# 关闭驱动程序
driver.close()

解释

  1. 连接到Neo4j数据库:使用GraphDatabase.driver创建Neo4j驱动程序实例,并提供数据库地址和认证信息。
  2. 读取CSV文件:使用Python内置的csv模块读取CSV文件。
  3. 创建节点和关系
    • 使用MERGE命令确保节点唯一性。如果节点不存在,则创建它;如果存在,则更新其属性。
    • 使用MATCHMERGE命令创建关系。如果关系不存在,则创建它。
  4. 关闭驱动程序:完成数据导入后,关闭驱动程序以释放资源。

注意事项

  • 确保Neo4j数据库正在运行,并且可以通过提供的URI访问。
  • 根据你的CSV文件内容和需求调整Cypher查询。
  • 对于大规模数据导入,建议使用neo4j-admin import工具,因为它比逐行插入更高效。

这样,你就可以使用Python脚本将CSV数据导入到Neo4j数据库中了。

版权声明:

本网仅为发布的内容提供存储空间,不对发表、转载的内容提供任何形式的保证。凡本网注明“来源:XXX网络”的作品,均转载自其它媒体,著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处。

我们尊重并感谢每一位作者,均已注明文章来源和作者。如因作品内容、版权或其它问题,请及时与我们联系,联系邮箱:809451989@qq.com,投稿邮箱:809451989@qq.com

热搜词