Scala / Spark：如何检查数据框是否包含特定的列列表？

Question

How can I test if my DF contains a list of columns? I tried this:

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

But I think contains method can only accept one column and not a list. Is there any other way to do this?

Answer 1

Try:

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

Or:

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

Answer 2

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

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