Определить функцию в Scala, используя тип DataFrame

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