在scala中利用ratings.csv和movies.csv,编程实现RDD转Data f r a me

时间: 2024-02-01 07:16:11 浏览: 23
在 Scala 中,可以使用 Spark 中的 `SparkSession` 对象将 RDD 转换为 DataFrame。假设我们有以下的 `ratings.csv` 和 `movies.csv` 文件: ``` // ratings.csv userId,movieId,rating,timestamp 1,1,4.0,964982703 1,3,4.0,964981247 1,6,4.0,964982224 1,47,5.0,964983815 1,50,5.0,964982931 2,1,3.0,964981247 2,3,1.0,964982224 2,6,2.0,964983815 2,47,4.0,964982931 2,50,3.0,964982703 // movies.csv movieId,title,genres 1,Toy Story (1995),Adventure|Animation|Children|Comedy|Fantasy 2,Jumanji (1995),Adventure|Children|Fantasy 3,Grumpier Old Men (1995),Comedy|Romance 4,Waiting to Exhale (1995),Comedy|Drama|Romance 5,Father of the Bride Part II (1995),Comedy 6,Heat (1995),Action|Crime|Thriller ``` 我们可以通过以下代码将它们转换为 DataFrame: ```scala import org.apache.spark.sql.{SparkSession, Row} import org.apache.spark.sql.types._ // 创建 SparkSession 对象 val spark = SparkSession.builder() .appName("RDD to DataFrame") .master("local[*]") .getOrCreate() // 读取 ratings.csv 文件 val ratingsRDD = spark.sparkContext.textFile("ratings.csv") .map(_.split(",")) .map(row => Row(row(0).toInt, row(1).toInt, row(2).toDouble, row(3).toLong)) // 定义 ratings 的 schema val ratingsSchema = StructType( Seq( StructField("userId", IntegerType, true), StructField("movieId", IntegerType, true), StructField("rating", DoubleType, true), StructField("timestamp", LongType, true) ) ) // 将 RDD 转换为 DataFrame val ratingsDF = spark.createDataFrame(ratingsRDD, ratingsSchema) // 读取 movies.csv 文件 val moviesRDD = spark.sparkContext.textFile("movies.csv") .map(_.split(",")) .map(row => Row(row(0).toInt, row(1), row(2))) // 定义 movies 的 schema val moviesSchema = StructType( Seq( StructField("movieId", IntegerType, true), StructField("title", StringType, true), StructField("genres", StringType, true) ) ) // 将 RDD 转换为 DataFrame val moviesDF = spark.createDataFrame(moviesRDD, moviesSchema) ``` 上述代码中,我们首先创建了一个 `SparkSession` 对象,然后使用 `textFile` 方法读取 `ratings.csv` 和 `movies.csv` 文件,并将每一行数据转换为 `Row` 对象。接着,我们定义了 `ratings` 和 `movies` 的 schema,然后使用 `createDataFrame` 方法将 RDD 转换为 DataFrame。最终,我们得到了 `ratingsDF` 和 `moviesDF` 两个 DataFrame 对象。

相关推荐

最新推荐

recommend-type

基于Android Studio的个人记账应用帮助用户轻松管理和跟踪他们的财务源码.zip

基于Android Studio的个人记账应用帮助用户轻松管理和跟踪他们的财务源码.zip基于Android Studio的个人记账应用帮助用户轻松管理和跟踪他们的财务源码.zip基于Android Studio的个人记账应用帮助用户轻松管理和跟踪他们的财务源码.zip基于Android Studio的个人记账应用帮助用户轻松管理和跟踪他们的财务源码.zip基于Android Studio的个人记账应用帮助用户轻松管理和跟踪他们的财务源码.zip基于Android Studio的个人记账应用帮助用户轻松管理和跟踪他们的财务源码.zip基于Android Studio的个人记账应用帮助用户轻松管理和跟踪他们的财务源码.zip基于Android Studio的个人记账应用帮助用户轻松管理和跟踪他们的财务源码.zip基于Android Studio的个人记账应用帮助用户轻松管理和跟踪他们的财务源码.zip基于Android Studio的个人记账应用帮助用户轻松管理和跟踪他们的财务源码.zip
recommend-type

基于python实现树莓派和传感器的植物生长环境评估信息系统

【作品名称】:基于python实现树莓派和传感器的植物生长环境评估信息系统 【适用人群】:适用于希望学习不同技术领域的小白或进阶学习者。可作为毕设项目、课程设计、大作业、工程实训或初期项目立项。 【项目介绍】:基于python实现树莓派和传感器的植物生长环境评估信息系统
recommend-type

优质资源,Yearning Mysql SQL审核平台

Yearning Mysql SQL审核平台(使用go语言)
recommend-type

c语言课程设计-职工资源管理系统.rar

void displayMenu() { printf("\n+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++\n"); printf("+ 职工资源管理系统 +\n"); printf("+ +\n"); printf("+ 1. 录入职工信息 +\n"); printf("+ 2. 显示全部职工信息 +\n"); printf("+ 3. 根据工号查询
recommend-type

华为OD机试D卷 - 来自异国的客人 - 免费看解析和代码.html

私信博主免费获取真题解析以及代码
recommend-type

zigbee-cluster-library-specification

最新的zigbee-cluster-library-specification说明文档。
recommend-type

管理建模和仿真的文件

管理Boualem Benatallah引用此版本:布阿利姆·贝纳塔拉。管理建模和仿真。约瑟夫-傅立叶大学-格勒诺布尔第一大学,1996年。法语。NNT:电话:00345357HAL ID:电话:00345357https://theses.hal.science/tel-003453572008年12月9日提交HAL是一个多学科的开放存取档案馆,用于存放和传播科学研究论文,无论它们是否被公开。论文可以来自法国或国外的教学和研究机构,也可以来自公共或私人研究中心。L’archive ouverte pluridisciplinaire
recommend-type

实现实时数据湖架构:Kafka与Hive集成

![实现实时数据湖架构:Kafka与Hive集成](https://img-blog.csdnimg.cn/img_convert/10eb2e6972b3b6086286fc64c0b3ee41.jpeg) # 1. 实时数据湖架构概述** 实时数据湖是一种现代数据管理架构,它允许企业以低延迟的方式收集、存储和处理大量数据。与传统数据仓库不同,实时数据湖不依赖于预先定义的模式,而是采用灵活的架构,可以处理各种数据类型和格式。这种架构为企业提供了以下优势: - **实时洞察:**实时数据湖允许企业访问最新的数据,从而做出更明智的决策。 - **数据民主化:**实时数据湖使各种利益相关者都可
recommend-type

用 Python 画一个可以动的爱心

好的,我可以帮助您使用Python来绘制一个动态的爱心。您可以使用turtle库来实现。以下是实现代码: ```python import turtle import math # 设置画布和画笔 canvas = turtle.Screen() canvas.bgcolor("black") pencil = turtle.Turtle() pencil.speed(0) pencil.color("red", "pink") pencil.pensize(3) # 定义爱心函数 def draw_love(heart_size, x_offset=0, y_offset=0):
recommend-type

JSBSim Reference Manual

JSBSim参考手册,其中包含JSBSim简介,JSBSim配置文件xml的编写语法,编程手册以及一些应用实例等。其中有部分内容还没有写完,估计有生之年很难看到完整版了,但是内容还是很有参考价值的。