使用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