正常的读写csv文件操作

val df=spark.read
            .format("csv")
            .option("header",true)              //文件是否包含表头
            .csv("D:\\test\\testfile.csv")      //文件地址
            .show()                             //展现数据

但是可能会出现中文字符乱码问题,所以需要进行优化

import spark.implicits._

val linesDF=spark.sparkContext
                 .hadoopFile("D:\\test\\testfile.csv", classOf[TextInputFormat], classOf[LongWritable], classOf[Text])
                 .map(pair => new String(pair._2.getBytes, 0, pair._2.getLength, "GBK"))
                 .mapPartitions(ite => ite.drop(1))        //删除第一行数据(表头数据)
                 .map(lines => lines.replace("\"",""))     //替换数据中的双引号
                 .map(lines => lines.split(","))           //切分数据
                 .map(s => (s(0),s(1),s(2),s(3),s(4)))     //转成Row类型
                 .toDF("id","姓名","性别","年龄","记录时间") 
                 .show()                        

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐