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

Question

我有一个简单的问题... 我尝试使用一个将datframe作为参数并返回另一个dataframe的函数

我这样做：

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