Scala / Spark: Как проверить, содержит ли фрейм данных СПЕЦИФИЧЕСКИЙ список столбцов?

Question

Как я могу проверить, содержит ли мой DF список столбцов? Я попробовал это:

finalDF.columns.contains(List("a", "b", "c"))

Но я думаю, что метод содержит может принимать только один столбец, а не список. Есть ли другой способ сделать это?

Answer 1

val colList = List("col1","col2","col3")

dataFrame.columns.filter(x => colList.contains(x))

Answer 2

Попробуйте:

List("a", "b", "c").forall(finalDF.columns.contains)

Или:

List("a", "b", "c").diff(finalDF.columns).isEmpty