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

>>> url = "file:///usr/local/spark/mycode/rdd/chapter4-data01.txt"

>>> rdd = spark.sparkContext.textFile(url).map(lambda line:line.split(','))

>>> rdd.take(3)

[['Aaron', 'OperatingSystem', '100'], ['Aaron', 'Python', '50'], ['Aaron', 'ComputerNetwork', '30']]

>>> 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操作的实现进行对比:

  • 每个分数+5分。

>>> df_scs.select('name','course',df_scs.score+5).show()

 

 

  • 总共有多少学生?

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

 

 

  • 总共开设了哪些课程?

>>> df_scs.select('course').distinct().show()

 

 

  • 每个学生选修了多少门课?

>>> df_scs.groupBy('name').count().show()

 

 

  • 每门课程有多少个学生选?

>>> df_scs.groupBy('course').count().show()

 

 

 

 

 

 

  • 每门课程大于95分的学生人数?

>>> df_scs.filter(df_scs.score>95).groupBy('course').count().show()

 

 

  • Tom选修了几门课?每门课多少分?

>>> df_scs.filter(df_scs.name=='Tom').show()

 

 

 

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

>>> df_scs.filter(df_scs.name=='Tom').orderBy(df_scs.score).show()

 

  • Tom的平均分。

>>> df_scs.filter(df_scs.name=='Tom').agg({"score":"mean"}).show()

 

  • 求每门课的平均分,最高分,最低分。

>>> df_scs.groupBy('course').avg('score').show()

  >>> df_scs.groupBy('course').max('score').show()

  >>> df_scs.groupBy('course').min('score').show()

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

>>> from pyspark.sql.functions import *
>>> df_scs.select(countDistinct('name').alias('学生人数'),countDistinct('course').alias('课程数'),round(mean('score'),2).alias('所有课的平均分')).show()

  • 每门课的不及格人数,通过率

>>> df_scs.filter(df_scs.score<60).groupBy('course').count().show()

 

 

 

  • 结果可视化。

 

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

fields = [StructField(...), ...]

schema = StructType(fields)

 类型:http://spark.apache.org/docs/latest/sql-ref-datatypes.html 

from pyspark.sql import Row

data = rdd.map(lambda p: Row(...))

 

Spark SQL DataFrame 操作

df.show()

df.printSchema()

df.count()

df.head(3)

df.collect()

df[‘name’]

df.name

df.first().asDict()

df.describe().show()

df.distinct()

df.filter(df['age'] > 21).show()

df.groupBy("age").count().show()

df.select('name', df['age‘] + 1).show()

df_scs.groupBy("course").avg('score').show()

 df_scs.agg({"score": "mean"}).show()

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

posted on 2021-05-20 19:16  wwxtl  阅读(60)  评论(0编辑  收藏  举报