Corrected fig9a taskcount
This commit is contained in:
parent
bfeff47d16
commit
cc7d904e83
1 changed files with 6 additions and 5 deletions
|
@ -31,8 +31,8 @@ spark = pyspark.sql.SparkSession.builder \
|
||||||
sc = spark.sparkContext
|
sc = spark.sparkContext
|
||||||
|
|
||||||
# READING INSTANCE EVENTS DATA
|
# READING INSTANCE EVENTS DATA
|
||||||
#dfepath = sys.argv[4] + "/" + cluster + "/" + cluster + "_instance_events*.json.gz"
|
dfepath = sys.argv[4] + "/" + cluster + "/" + cluster + "_instance_events*.json.gz"
|
||||||
dfepath = sys.argv[4] + "/" + cluster + "/" + cluster + "_test.json"
|
#dfepath = sys.argv[4] + "/" + cluster + "/" + cluster + "_test.json"
|
||||||
df = spark.read.json(dfepath)
|
df = spark.read.json(dfepath)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
|
@ -41,11 +41,12 @@ except:
|
||||||
df = df.withColumn("collection_type", lit(None).cast(ByteType()))
|
df = df.withColumn("collection_type", lit(None).cast(ByteType()))
|
||||||
|
|
||||||
df = df.rdd.filter(lambda x: x.collection_id is not None and x.instance_index is not None and
|
df = df.rdd.filter(lambda x: x.collection_id is not None and x.instance_index is not None and
|
||||||
(x.collection_type == 0)) \
|
(x.collection_type == 0 or x.collection_type is None)) \
|
||||||
.map(lambda x: (x.collection_id, x.instance_index)) \
|
.map(lambda x: (x.collection_id, x.instance_index)) \
|
||||||
.groupBy(lambda x: x[0]) \
|
.groupBy(lambda x: x[0]) \
|
||||||
.mapValues(lambda x: len(x))
|
.map(lambda x: [x[0], len(x[1])]) \
|
||||||
|
.toDF(["jobid", "task_count"])
|
||||||
|
|
||||||
df.write.parquet("/home/claudio/raid0/figure-9a-task-count-" + cluster + ".parquet")
|
df.write.parquet("/home/claudio/google_2019/thesis_queries/figure_9/figure-9a-task-count-" + cluster + ".parquet")
|
||||||
|
|
||||||
# vim: set ts=4 sw=4 et tw=120:
|
# vim: set ts=4 sw=4 et tw=120:
|
||||||
|
|
Loading…
Reference in a new issue