读学生课程分数文件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()