scala spark dataframe 修改字段类型

最新推荐文章于 2024-05-14 10:40:59 发布

slf_lu生生

最新推荐文章于 2024-05-14 10:40:59 发布

阅读量482

点赞数

文章标签： spark scala 大数据

本文链接：https://blog.csdn.net/qq_44920169/article/details/129710494

版权

scala spark dataframe 修改字段类型

package classifierAlg
import config.conf.{sc, spark_session}
import config.conf.spark_session.implicits._
import org.apache.spark.sql.functions._
import org.apache.spark.sql.types._
import org.apache.spark.sql.{Column, DataFrame}
object euclideanDist {
  def main(args: Array[String]): Unit = {
    val path:String = "data/irsdf/part-00000-ca2d6ce7-bcd0-4c24-aba9-e8cb01dcc04c-c000.csv"
    val df: DataFrame = spark_session.read.csv(path).toDF("ft1","ft2","ft3","ft4","label")
    val cols: Array[String] = df.columns
    val n: Int = cols.length
//     val colNames : Array[String] = df.schema.fieldNames
    val colsd: Array[Column] = cols.map(f => df(f).cast(DoubleType))
    val df2: DataFrame = df.select(colsd: _*)
    df2.printSchema()
 
  }
}