我会为 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))
提前致谢。