使用Spark SQL进行电信用户行为分析:
- 导入必要的库
from pyspark import SparkContext, SparkConf
from pyspark.sql import SparkSession
- 创建SparkSession
conf = SparkConf().setAppName("Telecom User Behavior Analysis")
sc = SparkContext(conf=conf)
spark = SparkSession(sc)
- 读取数据
# 读取用户行为数据
user_behavior = spark.read.format("csv").option("header", "true").load("user_behavior.csv")
# 读取用户信息数据
user_info = spark.read.format("csv").option("header", "true").load("user_info.csv")
- 数据清洗
# 将用户行为数据进行清洗
user_behavior = user_behavior.dropDuplicates() # 去重
user_behavior = user_behavior.dropna() # 去除缺失值
# 将用户信息数据进行清洗
user_info = user_info.dropDuplicates() # 去重
user_info = user_info.dropna() # 去除缺失值
- 数据预处理
# 将用户行为数据进行预处理,提取有用信息
user_behavior.createOrReplaceTempView("user_behavior")
user_behavior = spark.sql("""
SELECT user_id,
CASE WHEN behavior_type = '1' THEN 'click'
WHEN behavior_type = '2' THEN 'collect'
WHEN behavior_type = '3' THEN 'cart'
ELSE 'buy' END AS behavior_type,
COUNT(*) AS count
FROM user_behavior
GROUP BY user_id, behavior_type
ORDER BY user_id, behavior_type
""")
# 将用户信息数据进行预处理,提取有用信息
user_info.createOrReplaceTempView("user_info")
user_info = spark.sql("""
SELECT user_id,
CASE WHEN age <= '18' THEN '1'
WHEN age <= '30' THEN '2'
WHEN age <= '40' THEN '3'
ELSE '4' END AS age,
CASE WHEN gender = '0' THEN 'female'
ELSE 'male' END AS gender
FROM user_info
""")
- 数据分析
# 使用Spark SQL进行数据分析
user_behavior.createOrReplaceTempView("user_behavior")
user_info.createOrReplaceTempView("user_info")
result = spark.sql("""
SELECT ui.age,
ui.gender,
ub.behavior_type,
SUM(ub.count) AS count
FROM user_behavior ub
JOIN user_info ui ON ub.user_id = ui.user_id
GROUP BY ui.age, ui.gender, ub.behavior_type
ORDER BY ui.age, ui.gender, ub.behavior_type
""")
# 显示结果
result.show()