使用 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 作为我的函数的类型？

Answer 1

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

import org.apache.spark.sql.DataFrame