Spark数据统计与可视化:RDD、DataFrame和Spark SQL实战指南

Spark数据统计与可视化教程:RDD、DataFrame和Spark SQL实战,IPython Notebook交互式开发,数据处理与可视化技巧。

原文标题:大数据系列之Spark SQL、DataFrame和RDD数据统计与可视化

原文作者:牧羊人的方向

冷月清谈:

本文介绍了如何使用IPython Notebook进行交互式Spark程序开发,以及如何利用RDD、DataFrame和Spark SQL三种方式进行数据统计和可视化。

首先,文章详细讲解了IPython Notebook的安装和配置,包括创建远程连接密码、生成配置文件以及在IPython Notebook中使用Spark。之后,文章比较了RDD、DataFrame和Spark SQL三种数据结构的特点,并列出了它们在数据统计中常用的方法。

接下来,文章以用户数据为例,演示了如何创建RDD、DataFrame和Spark SQL,并分别使用这三种方式进行了数据筛选、排序、去重、分组统计和联接操作等常见的数据统计任务。最后,文章还介绍了如何将Spark DataFrame转换为Pandas DataFrame,并使用matplotlib库进行直方图和饼图的可视化展示。

怜星夜思:

1、文章提到了RDD、DataFrame和Spark SQL三种数据结构,它们各自的优缺点是什么?在实际应用中该如何选择?
2、文章中使用了IPython Notebook进行Spark程序开发,除了交互式界面外,它还有哪些优势?
3、文章最后提到了使用Pandas DataFrame进行数据可视化,除了matplotlib外,还有哪些Python库可以用来做数据可视化?它们各自有什么特点?

原文内容

Spark大数据分析中涉及到RDD、Data Frame和SparkSQL的操作,本文简要介绍三种方式在数据统计中的算子使用。

1、在IPython Notebook运行Python Spark程序

IPython Notebook具备交互式界面,可以在Web界面输入Python命令后立刻看到结果,还可将数据分析的过程和运行后的命令与结果存储成笔记本,下次可以打开笔记本,重新执行这些命令,IPython Notebook笔记本可以包含文字、数学公式、程序代码、结果、图形等。

1.1 安装IPython

1)若无gcc,需先安装gcc

[root@tango-spark01 /]# gcc v
[root@tango-spark01 /]# yum install gcc

2)若无pip,安装pip

[root@tango-spark01 /]# pip v
[root@tango-spark01 /]# wget https://bootstrap.pypa.io/get-pip.py --no-check-certificate

3)安装Python开发包

[root@tango-spark01 /]# yum install python-devel

4)执行以下命令安装IPython和IPython Notebook:

[root@tango-spark01 /]# pip install ipython
[root@tango-spark01 /]# pip install urllib3
[root@tango-spark01 /]# pip install jupyter

5)输入ipython进入交互界面

6)输入jupyter notebook

1.2 IPython配置

1)创建远程连接密码

In [2]: from notebook.auth import passwd;
In [3]: passwd()
Enter password:
Verify password:
Out[3]: 'sha1:fea9052f4d92:114eb32c684004bf4a6196faac0b26a28156fe5d'

2)生成jupyter配置文件

[root@tango-spark01 /]# jupyter notebook --generate-config
Writing default config to: /root/.jupyter/jupyter_notebook_config.py

3)打开配置文件,设置以下内容

## The IP address the notebook server will listen on.
#c.NotebookApp.ip = 'localhost'
c.NotebookApp.ip = '0.0.0.0'
## The directory to use for notebooks and kernels.
#c.NotebookApp.notebook_dir = u''
c.NotebookApp.notebook_dir = u'/usr/local/spark/ipynotebook'
## Hashed password to use for web authentication.
# To generate, type in a python/IPython shell:
# from notebook.auth import passwd; passwd()
# The string should be of the form type:salt:hashed-password.
#c.NotebookApp.password = u''
c.NotebookApp.password = u'sha1:fea9052f4d92:114eb32c684004bf4a6196faac0b26a28156fe5d'

4)打开jupyter notebook

[root@tango-spark01 /]# jupyter notebook --allow-root
[I 14:20:05.618 NotebookApp] Serving notebooks from local directory: /usr/local/spark/ipynotebook
[I 14:20:05.618 NotebookApp] The Jupyter Notebook is running at:
[I 14:20:05.619 NotebookApp] http://(tango-spark01 or 127.0.0.1):8888/
[I 14:20:05.619 NotebookApp] Use Control-C to stop this server and shut down all kernels (twice to skip confirmation).
[W 14:20:05.619 NotebookApp] No web browser found: could not locate runnable browser.
[I 14:21:00.346 NotebookApp] 302 GET / (192.168.112.1) 2.50ms
[I 14:21:00.352 NotebookApp] 302 GET /tree? (192.168.112.1) 1.71ms
[I 14:22:16.241 NotebookApp] 302 POST /login?next=%2Ftree%3F (192.168.112.1) 1.58ms

5)浏览器输入地址和端口

输入密码登录进去

1.3 在IPython Notebook中使用Spark

1)进入ipynotebook工作目录

[root@tango-spark01 /]# cd /usr/local/spark/ipynotebook
[root@tango-spark01 ipynotebook]#

2)在IPython Notebook界面中运行pyspark

[root@tango-spark01 ipynotebook]# PYSPARK_DRIVER_PYTHON=jupyter PYSPARK_DRIVER_PYTHON_OPTS="notebook"  pyspark

3)单击New选择Python 2,新建Notebook

4)新建Notebook后会出现新的页面,默认notebook名称为Untitled,单击后修改名称

5)在Notebook运行程序代码

6)保存Notebook下次可继续打开使用

2、Spark SQL、DataFrame、RDD数据统计与可视化
2.1 RDD、DataFrame和Spark SQL比较

RDD和Data Frame都是Spark平台下分布式弹性数据集,都有惰性机制,在进行创建、转换时不会立即执行,等到Action时才会遍历运算。

  1. RDD API进行数据统计,主要使用map配合reduceByKey,需要有Map/Reduce概念

  2. 与RDD不同的是Data Frame更像是传统的数据库表格,除了数据以外,还记录了数据的结构信息

  3. Spark SQL则是由DataFrame派生出来,必须先创建DataFrame,然后通过登录Spark SQL temp table就可以使用Spark SQL语句,直接使用SQL语句进行查询

下表列出在进行数据统计计算时候,RDD、Data Frame和Spark SQL使用的不同方法。

Items 功能描述
RDD API userRDD.map(lambda x:(x[2],1)).reduceByKey(lambda x,y: x+y).collect()
DataFrame user_df.select(“gender”).groupby(“gender”).count().show()
Spark SQL sqlContext.sql(“””SELECT gender,count(*) counts FROM user_table GROUP BY gender”””).show()
2.2 创建RDD、DataFrame和Spark SQL

在Hadoop YARN-client模式运行IPython Notebook

[root@tango-spark01 ipynotebook]# PYSPARK_DRIVER_PYTHON=jupyter PYSPARK_DRIVER_PYTHON_OPTS="notebook"  HADOOP_CONF_DIR=/usr/local/spark/hadoop-2.9.0/etc/hadoop  pyspark --master yarn --deploy-mode client
  • 创建RDD

1)配置文件读取路径

global Path
if sc.master[0:5] =="local":
Path="file:/usr/local/spark/ipynotebook/"
else:
Path="hdfs://tango-spark01:9000/input/"
  1. 如果sc.master[0:5]是“local”,代表当前在本地运行,读取本地文件

  2. 如果sc.master[0:5]不是“local”,有可能是YARN client或Spark Stand Alone,必须读取HDFS文件

2)读取文本文件并且查看数据项数

RawUserRDD=sc.textFile(Path+"data/u.user")
RawUserRDD.count()
RawUserRDD.take(5)

3)获取每一个字段

userRDD= RawUserRDD.map(lambda line:line.split("|"))
userRDD.take(5)

  • 创建Data Frame

1)创建sqlContext:在Spark早期版本中,spark context是Spark的入口、SQLContext是SQL入口、HiveContext是hive入口。在Spark 2.0中,使用Spark Session可同时具备spark context、sqlContext、HiveContext功能

sqlContext=SparkSession.builder.getOrCreate()

2)定义Schema:定义DataFrames的每个字段名与数据类型

from pyspark.sql import Row
user_Rows = userRDD.map(lambda p:
Row(userid=int(p[0]),age=int(p[1]),gender=p[2],occupation=p[3],zipcode=p[4]))
user_Rows.take(5)

3)创建DataFrames:使用sqlContext.createDataFrame()方法创建DataFrame

user_df=sqlContext.createDataFrame(user_Rows)
user_df.printSchema()

4)查看DataFrames数据

user_df.show(5)

5)为DataFrame创建别名:可以使用.alias帮DataFrame创建别名

df=user_df.alias("df")
df.show(5)

  • 使用SparkSQL

创建DataFrame后,使用该DataFrame登录Spark SQL temp table,登录后可以使用Spark SQL

1)登录临时表

user_df.registerTempTable("user_table")

2)使用Spark SQL查看项数

sqlContext.sql("SELECT count(*) counts FROM user_table").show()

3)多行输入Spark SQL语句,需要使用3个双引号引住SQL

sqlContext.sql("""
SELECT count(*) counts
FROM user_table
"""
).show()

4)使用SparkSQL查看数据,限定数据项

sqlContext.sql("SELECT * FROM user_table").show()
sqlContext.sql("SELECT * FROM user_table").show(5)
sqlContext.sql("SELECT * FROM user_table LIMIT 5").show()

2.3 数据统计操作
2.3.1 筛选数据
  • 使用RDD筛选数据

RDD中使用filter方法筛选每一项数据,配合lambda语句创建匿名函数传入参数

userRDD.filter(lambda r:r[3]=='technician' and r[2]=='M' and r[1]=='24').take(5)
  • 输入DataFrames筛选数据

user_df.filter((df.occupation=='technician')&(df.gender=='M')&(df.age==24)).show()
  • 使用Spark SQL筛选数据

sqlContext.sql("""
SELECT *
FROM user_table
where occupation='technician' and gender='M' and age=24"""
).show(5)
2.3.2 按字段给数据排序
  • RDD按字段给数据排序

userRDD.takeOrdered(5,key=lambda x:int(x[1]))——升序排序
userRDD.takeOrdered(5,key=lambda x:-1*int(x[1]))——降序排序
userRDD.takeOrdered(5,key=lambda x:(-int(x[1]),x[2]))——多个字段排序
  • 使用DataFrame排序

user_df.select("userid","occupation","gender","age").orderBy("age").show(5)——升序
user_df.select("userid","occupation","gender","age").orderBy("age",ascending=0).show(5)
df.orderBy(["age","gender"],ascending=[0,1]).show(5)——多个字段排序
  • 使用Spark SQL排序

sqlContext.sql("""
SELECT userid,occupation,gender,age FROM user_table
order by age desc,gender"""
).show(5)
2.3.3 显示不重复数据
  • RDD显示不重复数据

userRDD.map(lambda x:x[2]).distinct().collect()
  • DataFrame显示不重复数据

user_df.select("gender").distinct().show()
  • Spark SQL显示不重复数据

sqlContext.sql("select distinct gender FROM user_table").show()
2.3.4 分组统计数据

1)RDD分组统计数据

userRDD.map(lambda x:(x[2],1)).reduceByKey(lambda x,y:x+y).collect()

2)DataFrames分组统计数据

user_df.select("gender").groupby("gender").count().show()

3)Spark SQL分组统计数据

sqlContext.sql("""
SELECT gender,count(*) counts FROM user_table
group by gender"""
).show()
2.3.5 Join联接数据
  • 准备zipcode数据

1)拷贝数据到HDFS目录下

[root@tango-spark01 data]# hadoop fs -copyFromLocal -f /usr/local/spark/ipynotebook/data/free-zipcode-database-Primary.csv  /input/data

2)读取并查看数据

Path="hdfs://tango-spark01:9000/input/"
rawDataWithHeader=sc.textFile(Path+"data/free-zipcode-database-Primary.csv")
rawDataWithHeader.take(5)

3)删除第一项数据

header = rawDataWithHeader.first()
rawData = rawDataWithHeader.filter(lambda x:x !=header)

4)删除特殊符号

rawData.first()
rData=rawData.map(lambda x:x.replace("\"",""))
rData.first()

5)获取每一个字段

zipRDD=rData.map(lambda x:x.split(","))
zipRDD.first()
  • 创建zipcode_tab

1)创建zipCode Row的schema

from pyspark.sql import Row
zipcode_data = zipRDD.map(lambda p:
Row(zipcode=int(p[0]),zipCodeType=p[1],city=p[2],state=p[3]))
zipcode_data.take(5)

2)Row类型数据创建DataFrames

zipcode_df=sqlContext.createDataFrame(zipcode_data)
zipcode_df.printSchema()

3)创建登录临时表

zipcode_df.registerTempTable("zipcode_table")
zipcode_df.show(10)
  • Spark SQL联接zipcode_table

sqlContext.sql("""
select u.*,z.city,z.state from user_table u
left join zipcode_table z ON u.zipcode=z.zipcode
where z.state='NY'
"""
).show(10)

2.3.6 使用Pandas DataFrame绘图
  • 按照不同的州统计并以直方图显示

1)转换为Pandas DataFrames

import pandas as pd
GroupByState_pandas_df = GroupByState_df.toPandas().set_index('state')
GroupByState_pandas_df

2)使用Pandas DataFrames绘出直方图

import matplotlib.pyplot as plt
#matplotlib inline
ax=GroupByState_pandas_df['count'].plot(kind='bar',title='State',figsize=(12,6),legend=True,fontsize=12)
plt.show()

  • 按照不同的职业统计并以饼图显示

1)创建Occupation_df

Occupation_df=sqlContext.sql("""
SELECT u.occupation,count(*) counts
FROM user_table u
GROUP
BY occupation
""")
Occupation_df.show(5)

2)创建Occupation_pandas_df

Occupation_pandas_df=Occupation_df.toPandas().set_index('occupation')
Occupation_pandas_df

3)用Pandas DataFrame是绘出饼图PieChart

ax=Occupation_pandas_df['counts'].plot(kind='pie',
title='occupation',figsize=(8,8),startangle=90,autopct='%1.1f%%')
ax.legend(bbox_to_anchor=(1.05,1),loc=2,borderaxespad=0.)
plt.show()
  1. kind='pie':绘制饼图

  2. startangle=90:设置图形旋转角度

  3. autopct='%1.1f%%':设置显示饼图%

参考资料

  1. Python+Spark 2.0+Hadoop机器学习与大数据实战,林大贵


如果想做3D可视化,可以考虑Mayavi或者Vispy。如果想做更炫酷的可视化,可以试试PyOpenGL或者VPython。

我觉得IPython Notebook最大的优势是可以将代码、结果和文档整合在一起,方便分享和交流。比如做数据分析报告,可以直接把代码、图表和解释都放在一个Notebook里,非常直观。

关于RDD、DataFrame和Spark SQL的选择,我觉得要看具体场景。RDD比较底层,控制力强,适合需要自定义操作的情况。DataFrame和Spark SQL更高级一些,自带优化,适合常规的数据处理和分析。如果熟悉SQL,Spark SQL用起来会更方便。

RDD就像手动挡,DataFrame和Spark SQL就像自动挡。手动挡可以精细控制,但比较麻烦;自动挡方便快捷,但灵活性差一些。所以,如果你需要极致性能或者自定义操作,就选RDD;如果只是常规的数据处理,DataFrame和Spark SQL就足够了,还能省不少力气。

除了Seaborn、Plotly和Bokeh,还有Altair,它是一个声明式可视化库,可以让你用更少的代码创建更丰富的图表。还有Folium,它可以用来创建地图可视化。

除了文章提到的,IPython Notebook还可以进行版本控制,方便团队协作。而且它有很多插件可以扩展功能,比如代码补全、语法高亮等,可以提高开发效率。

IPython Notebook支持多种语言,不仅仅是Python,还可以用R、Julia等。而且它可以方便地嵌入图片、视频和网页,让你的Notebook更加生动丰富。

从性能角度来看,RDD的性能最高,因为它直接操作底层数据。DataFrame次之,Spark SQL的性能略逊于DataFrame。但是,Spark SQL的使用最便捷,DataFrame的API也比较友好,RDD的API则比较底层,使用起来比较复杂。

Seaborn是基于matplotlib的高级可视化库,可以绘制更美观、更复杂的图表,而且使用起来更方便。Plotly可以绘制交互式图表,方便用户探索数据。Bokeh也是一个交互式可视化库,支持流式数据可视化。