scala - 使用 DataFrame 类型在 Scala 中定义函数

Question

我会有一个简单的问题...我尝试使用一个函数，该函数将一个数据帧作为参数并返回另一个数据帧

我愿意：

val get_nb_previous_offre : ((DataFrame) => (DataFrame)) = (arg1: DataFrame) => {
   // create new_df using arg1 (=DataFrame I pass in parameter)
      new_df    
} 

我得到错误：

<console>:32: error: not found: type DataFrame
   val get_nb_previous_offre : ((DataFrame) => (DataFrame)) = (arg1: DataFrame) => {

是否有我必须导入的库才能使用 DataFrame 作为我的函数的类型？

标签： scalaapache-spark

Answer 1

是的，您需要导入此类型：

import org.apache.spark.sql.DataFrame