Define function in scala using DataFrame type

Question

I would have a simple question... I try to use a function that ate a dataframe as argument and returns another dataframe

I do:

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

I get the error:

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

Is there a librairy I have to import to use DataFrame as a type for my function?

Answer 1

是的，您需要导入以下类型：

import org.apache.spark.sql.DataFrame