0

我会为 neo4j 写一个巨大的图表。使用我的代码只需不到两个月的时间。

我从Kaggle 的事件推荐挑战user_friends.csv中获取数据,我使用的文件看起来像

user,friends
3197468391,1346449342 3873244116 4226080662, ... 

我使用 py2neobatch工具来生成代码。这是我能做的最好的还是有其他方法可以显着减少运行时间?

这是代码

#!/usr/bin/env python

from __future__ import division
from time import time
import sqlite3
from py2neo import neo4j

graph = neo4j.GraphDatabaseService("http://localhost:7474/db/data/")
batch = neo4j.WriteBatch(graph)

people = graph.get_or_create_index( neo4j.Node,"people")
friends = graph.get_or_create_index( neo4j.Relationship,"friends")

con = sqlite3.connect("test.db")
c = con.cursor()
c.execute("SELECT user, friends FROM user_friends LIMIT 2;") 

t=time()
for u_f in c:
    u_node = graph.get_or_create_indexed_node("people",'name',u_f[0]) 

    for f in u_f[1].split(" "):
        f_node = graph.get_or_create_indexed_node("people",'name', f)

        if not f_node.is_related_to(u_node, neo4j.Direction.BOTH,"friends"):
            batch.create((u_node,'friends',f_node))

    batch.submit()
print time()-t

我也找不到使用高级py2neo设施创建无向图的方法?我知道cypher可以这样做create (node(1) -[:friends]-node(2))

提前致谢。

4

1 回答 1

1

您应该创建不与Direction.BOTH. 选择一个方向,然后Direction.BOTH在遍历时忽略使用它 - 它对性能没有影响,但关系方向是确定性的。当您说 . 时,Cypher 正是这样做的a--b

于 2013-05-03T13:37:35.610 回答