学生课程分数的Spark SQL分析

读学生课程分数文件chapter4-data01.txt，创建DataFrame。

url = "file:///D:/chapter4-data01.txt"
rdd = spark.sparkContext.textFile(url).map(lambda line:line.split(','))
rdd.take(3)

from pyspark.sql.types import IntegerType,StringType,StructField,StructType
from pyspark.sql import Row

#生成“表头”
fields = [StructField('name',StringType(),True),StructField('course',StringType(),True),StructField('score',IntegerType(),True)]
schema = StructType(fields)

# 生成“表中的记录”
data = rdd.map(lambda p:Row(p[0],p[1],int(p[2])))

# 把“表头”和“表中的记录”拼接在一起
df_scs = spark.createDataFrame(data,schema)
df_scs.printSchema()
df_scs.show()

一：用DataFrame的操作或SQL语句完成以下数据分析要求，并和用RDD操作的实现进行对比：

1.每个分数+5分。

# 1.每个分数+5分。
df_scs.select('name','course',df_scs.score+5).show()

2.总共有多少学生？

# 2.总共有多少学生？
df_scs.select(df_scs.name).distinct().count()

df_scs.select(df_scs['name']).distinct().count()

df_scs.select('name').distinct().count()

3.总共开设了哪些课程？

# 3.总共开设了哪些课程？
df_scs.select('course').distinct().show()

4.每个学生选修了多少门课？

# 4.每个学生选修了多少门课？
df_scs.groupBy('name').count().show()

5.每门课程有多少个学生选？

# 5.每门课程有多少个学生选？
df_scs.groupBy('course').count().show()

6.每门课程大于95分的学生人数？

# 6.每门课程大于95分的学生人数？
df_scs.filter(df_scs.score>95).groupBy('course').count().show()

7.Tom选修了几门课？每门课多少分？

# 7.Tom选修了几门课？每门课多少分？
df_scs.filter(df_scs.name=='Tom').show()

8.Tom的成绩按分数大小排序。

# 8.Tom的成绩按分数大小排序。
df_scs.filter(df_scs.name=='Tom').orderBy(df_scs.score).show()

9.Tom的平均分。

# 9.Tom的平均分。
df_scs.filter(df_scs.name=='Tom').agg({"score":"mean"}).show()

10.求每门课的平均分，最高分，最低分。

# 10.求每门课的平均分，最高分，最低分。
df_scs.groupBy("course").agg({"score": "mean"}).show()

df_scs.groupBy("course").agg({"score": "max"}).show()

df_scs.groupBy("course").agg({"score": "min"}).show()

11.求每门课的选修人数及平均分，精确到2位小数。

# 11.求每门课的选修人数及平均分，精确到2位小数。
df_scs.select(countDistinct('name').alias('学生人数'),countDistinct('course').alias('课程数'),round(mean('score'),2).alias('所有课的平均分')).show()

12.每门课的不及格人数，通过率

# 12.每门课的不及格人数，通过率
df_scs.filter(df_scs.score<60).groupBy('course').count().show()

13.结果可视化。

二、用SQL语句完成以上数据分析要求

1.每个分数+5分。

# 1.每个分数+5分。
spark.sql("SELECT name,course,score+5 FROM scs").show()

2.总共有多少学生？

# 2.总共有多少学生？
spark.sql("SELECT course,count(name) as n FROM scs group by course").show()

3.总共开设了哪些课程？

# 3.总共开设了哪些课程？
spark.sql("SELECT course FROM scs group by course").show()

4.每个学生选修了多少门课？

# 4.每个学生选修了多少门课？
spark.sql("SELECT name,count(course) FROM scs group by name").show() 

5.每门课程有多少个学生选？

# 5.每门课程有多少个学生选？
spark.sql("SELECT course,count(name) FROM scs group by course").show() 

6.每门课程大于95分的学生人数？

# 6.每门课程大于95分的学生人数？
spark.sql("SELECT course,count(name) FROM scs where score>95 group by course").show() 

7.Tom选修了几门课？每门课多少分？

# 7.Tom选修了几门课？每门课多少分？
spark.sql("SELECT * FROM scs where name=='Tom'").show()

8.Tom的成绩按分数大小排序。

# 8.Tom的成绩按分数大小排序。
spark.sql("SELECT * FROM scs where name=='Tom' order by score desc").show() 

9.Tom的平均分。

# 9.Tom的平均分。
spark.sql("SELECT avg(score) as avg FROM scs where name=='Tom'").show()

10.求每门课的平均分，最高分，最低分。

# 10.求每门课的平均分，最高分，最低分。
spark.sql("SELECT course,count(name) as n,avg(score) as avg,max(score) as max,min(score) as min FROM scs group by course").show() 

11.求每门课的选修人数及平均分，精确到2位小数。

# 11.求每门课的选修人数及平均分，精确到2位小数。
spark.sql("SELECT course,count(name) as n,round(avg(score),2) as avg FROM scs group by course").show() 

12.每门课的不及格人数，通过率

# 12.每门课的不及格人数，通过率
spark.sql("SELECT course,count(name) as n,avg(score) as avg FROM scs group by course").createOrReplaceTempView("a")
spark.sql("SELECT course,count(score) as notPass FROM scs where score<60 group by course").createOrReplaceTempView("b")
spark.sql("SELECT * from a left join b on a.course=b.course").show()

spark.sql("select a.course,round(a.avg,2),b.notPass,round((a.n-b.notPass)/a.n,2) as passRat from a left join b on a.course=b.course").show()

三、对比分别用RDD操作实现、用DataFrame操作实现和用SQL语句实现的异同。(比较两个以上问题)

例如：每门课的选修人数与平均分

1.RDD实现

# 方法一 combineByKey()
course = words.combineByKey(lambda v:(int(v),1),lambda c,v:(c[0]+int(v),c[1]+1),lambda c1,c2:(c1[0]+c2[0],c1[1]+c2[1]))  #每门课的选修人数及所有人的总分。combineByKey()
course.map(lambda x:(x[0],x[1][1],round(x[1][0]/x[1][1],2))).collect() #每门课的选修人数及平均分，精确到2位小数。

#方法二 reduceByKey()
lines.map(lambda line:line.split(',')).map(lambda x:(x[1],(int(x[2]),1))).reduceByKey(lambda a,b:(a[0]+b[0],a[1]+b[1])).collect()

2.DataFrame实现

df_scs.groupBy("course").agg({'name':'count','score':'mean'}).withColumnRenamed("avg(score)",'avg').withColumnRenamed("count(name)",'n').show()
df_scs.groupBy("course").avg('score').show()
df_scs.groupBy("course").count().show()

3.SQL语句

spark.sql("SELECT course,count(name) as n,avg(score) as avg FROM scs group by course").show()

四、结果可视化。

函数：http://spark.apache.org/docs/2.2.0/api/python/pyspark.sql.html#module-pyspark.sql.functions