How to convert column of arrays of strings to strings?

后端 未结 4 2049
死守一世寂寞
死守一世寂寞 2020-12-13 16:23

I have a column, which is of type array < string > in spark tables. I am using SQL to query these spark tables. I wanted to convert the array < s

4条回答
  •  旧巷少年郎
    2020-12-13 17:06

    In Spark 2.1+ to do the concatenation of the values in a single Array column you can use the following:

    1. concat_ws standard function
    2. map operator
    3. a user-defined function (UDF)

    concat_ws Standard Function

    Use concat_ws function.

    concat_ws(sep: String, exprs: Column*): Column Concatenates multiple input string columns together into a single string column, using the given separator.

    val solution = words.withColumn("codes", concat_ws(" ", $"rate_plan_code"))
    scala> solution.show
    +--------------+-----------+
    |         words|      codes|
    +--------------+-----------+
    |[hello, world]|hello world|
    +--------------+-----------+
    

    map Operator

    Use map operator to have full control of what and how should be transformed.

    map[U](func: (T) ⇒ U): Dataset[U] Returns a new Dataset that contains the result of applying func to each element.

    scala> codes.show(false)
    +---+---------------------------+
    |id |rate_plan_code             |
    +---+---------------------------+
    |0  |[AAA, RACK, SMOBIX, SMOBPX]|
    +---+---------------------------+
    
    val codesAsSingleString = codes.as[(Long, Array[String])]
      .map { case (id, codes) => (id, codes.mkString(", ")) }
      .toDF("id", "codes")
    
    scala> codesAsSingleString.show(false)
    +---+-------------------------+
    |id |codes                    |
    +---+-------------------------+
    |0  |AAA, RACK, SMOBIX, SMOBPX|
    +---+-------------------------+
    
    scala> codesAsSingleString.printSchema
    root
     |-- id: long (nullable = false)
     |-- codes: string (nullable = true)
    

提交回复
热议问题