Мотивы в писпарке GraphFrames - PullRequest
0 голосов
/ 25 мая 2018

Я новичок в pyspark и пытаюсь найти мотивы в GraphFrame.Я получаю пустые результаты, хотя я точно знаю, что существуют отношения между вершинами и ребрами.Я запускаю это с Spark 1.6 в Jupyter на Cloudera.Скриншоты моих вершин и ребер (а также схемы) находятся по следующим ссылкам: verticesRDD и dgeRDD

Я читаю GraphFrames , ноне понимаю ... У меня пока есть следующий код.Куда я иду не так ..?

#import relevant libraries for Graph Frames
from pyspark import SparkContext
from pyspark.sql import SQLContext
from pyspark.sql.functions import desc
from graphframes import *

#Read the csv files 
verticesRDD = sqlContext.read.format("com.databricks.spark.csv").options(header='true', inferschema='true').load("filepath/station.csv")
edgesRDD = sqlContext.read.format("com.databricks.spark.csv").options(header='true', inferschema='true').load("filepath/trip.csv")

#Renaming the id columns to enable GraphFrame 
verticesRDD = verticesRDD.withColumnRenamed("station_ID", "id")
edgesRDD = edgesRDD.withColumnRenamed("Trip ID", "id")
edgesRDD = edgesRDD.withColumnRenamed("Start Station", "src")
edgesRDD = edgesRDD.withColumnRenamed("End Station", "dst")

#Register as temporary tables for running the analysis
verticesRDD.registerTempTable("verticesRDD")
edgesRDD.registerTempTable("edgesRDD")
#Note: whether i register the RDDs as temp tables or not, i get the same epty results... so im not sure if this step is really needed

#Make the GraphFrame
g = GraphFrame(verticesRDD, edgesRDD)

print g
#this deisplays the following:
#GraphFrame(v:[id: int, name: string, lat: double, long: double, dockcount: int, landmark: string, installation: string], e:[src: string, dst: string, id: int, Duration: int, Start Date: string, Start Terminal: int, End Date: string, End Terminal: int, Bike #: int, Subscriber Type: string, Zip Code: string])

#Stations where a is connected to b
motifs = g.find("(a)-[e1]->(b)")
motifs.show()

+---+---+---+
| e1|  a|  b|
+---+---+---+
+---+---+---+

motifs = g.find("(a)-[e1]->(b); (b)-[e2]->(a)")
motifs.show()

+---+---+---+---+
| e1|  a|  b| e2|
+---+---+---+---+
+---+---+---+---+


motifs = g.find("(a)-[e1]->(b); (b)-[e2]->(c)")
motifs.show()

+---+---+---+---+---+
| e1|  a|  b| e2|  c|
+---+---+---+---+---+
+---+---+---+---+---+

#Stations where a is connected to b, b is connected to c 
#but c is not connected to a
motifs = g.find("(a)-[e1]->(b); (b)-[e2]->(c)").filter("(c!=a)")
motifs.show()

+---+---+---+---+---+
| e1|  a|  b| e2|  c|
+---+---+---+---+---+
+---+---+---+---+---+

1 Ответ

0 голосов
/ 30 мая 2018

Проблема заключалась в том, как я определял свои вершины.Я переименовал "station_id" в "id", когда на самом деле это должно было быть "name". Поэтому эта строка

verticesRDD = verticesRDD.withColumnRenamed("station_ID", "id")

должна быть

verticesRDD = verticesRDD.withColumnRenamed("name", "id")

Мотивы работают правильнос этим изменением!

...