当前位置：首页 > article >正文

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

article 2025/1/13 13:46:15

要将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()

解释

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