首页 > 解决方案 > WrappedArray$ofRef 不能转换为 scala.collection.immutable.Seq

问题描述

我正在尝试将一些 Python 代码转换为 Scala。

Python代码:

def col_c(o_row_ids,n_row_ids):
    o_set=set(o_row_ids)
    n_set=set(n_row_ids)
    if o_set=n_set
        return "in"
    elif o_set < n_set:
        return "Me"
    elif n_set < o_set:
        return "Sp"
    return "SM"

斯卡拉代码:

def col_c: UserDefinedFunction = udf((o_row_ids:Seq[String], n_row_ids: Seq[String]) => {
    val o_set = o_row_ids.toSet.count(z => true) //set(o_row_ids)
    val n_set = n_row_ids.toSet.count(z=> true)
    if (o_set == n_set)
        "In"
    else if( o_set < n_set)
        "Me"
    else if (n_set < o_set)
        "Sp"
    else "SM"
})

但我收到以下错误:

无法执行用户定义的函数(col_c(数组(字符串),数组(字符串)=>字符串 scala.collection.mutable.wrappedArray$ofRef 无法转换为 scala .collection.immutable.Seq

任何建议如何防止此错误?

标签: scalaapache-sparkpysparkuser-defined-functionsscala-collections

解决方案


WrappedArrayextendsscala.collection.mutable.Seq本身扩展scala.collection.Seq

看起来你已经导入scala.collection.immutable.Seq了,因此出现了错误。

解决您的问题的一种可能性是将您的 UDF 输入为scala.collection.Seq.


推荐阅读