数据分析之图算法: 基于Spark和Neo4j

96

｜

第

6

章

图 6-2：示例图模型

下面基于示例

CSV

文件在

Spark

和

Neo4j

中创建图。

6.1.1

　将数据导入

Spark

首先从

Spark

和

GraphFrames

包中导入所需的包：

from graphframes import

*

下面的函数从示例

CSV

文件中创建一个

GraphFrame

对象：

def

create_software_graph():

nodes = spark.read.csv("data/sw-nodes.csv", header=True)

relationships = spark.read.csv("data/sw-relationships.csv", header=True)

return

GraphFrame(nodes, relationships)

然后调用该函数。

g = create_software_graph()